Update app.py

app.py

@@ -114,12 +114,14 @@ def next_image(embs, ys):
 
 
 def start(_, embs, ys):
+    embs, ys = next_image(embs, ys)
     return [
             gr.Button(value='Like', interactive=True), 
             gr.Button(value='Neither', interactive=True), 
             gr.Button(value='Dislike', interactive=True),
             gr.Button(value='Start', interactive=False),
-            next_image(embs, ys)
+            embs,
+            ys,
             ]
 
 
@@ -128,11 +130,12 @@ def choose(choice, embs, ys):
         choice = 1
     elif choice == 'Neither':
         _ = embs.pop(-1)
-        return next_image(embs, ys)
+        embs, ys = next_image(embs, ys)
     else:
         choice = 0
     ys.append(choice)
-    return next_image(embs, ys)
+    embs, ys = next_image(embs, ys)
+    return {'img': image, 'embs': embs, 'ys': ys}
 
 css = "div#output-image {height: 768px !important; width: 768px !important; margin:auto;}"
 with gr.Blocks(css=css) as demo: