Update app.py

app.py

@@ -254,21 +254,23 @@ def background_next_image():
     
     user_emb = get_user_emb(embs, ys)
     img, embs = generate(user_emb)
-    tmp_df = pd.DataFrame(columns=['paths', 'embeddings', 'ips', 'user:rating', 'latest_user_to_rate'])
-    tmp_df['paths'] = [img]
-    tmp_df['embeddings'] = [embs]
-    tmp_df['user:rating'] = [{' ': ' '}]
-    prevs_df = pd.concat((prevs_df, tmp_df))
-    # we can free up storage by deleting the image
-    if len(prevs_df) > 50:
-        oldest_path = prevs_df.iloc[0]['paths']
-        if os.path.isfile(oldest_path):
-            os.remove(oldest_path)
-        else:
-            # If it fails, inform the user.
-            print("Error: %s file not found" % oldest_path)
-        # only keep 50 images & embeddings & ips, then remove oldest
-        prevs_df = prevs_df.iloc[1:]
+    print(img)
+    if img:
+        tmp_df = pd.DataFrame(columns=['paths', 'embeddings', 'ips', 'user:rating', 'latest_user_to_rate'])
+        tmp_df['paths'] = [img]
+        tmp_df['embeddings'] = [embs]
+        tmp_df['user:rating'] = [{' ': ' '}]
+        prevs_df = pd.concat((prevs_df, tmp_df))
+        # we can free up storage by deleting the image
+        if len(prevs_df) > 50:
+            oldest_path = prevs_df.iloc[0]['paths']
+            if os.path.isfile(oldest_path):
+                os.remove(oldest_path)
+            else:
+                # If it fails, inform the user.
+                print("Error: %s file not found" % oldest_path)
+            # only keep 50 images & embeddings & ips, then remove oldest
+            prevs_df = prevs_df.iloc[1:]
     
 
 def pluck_embs_ys(user_id):