Update app.py

app.py

@@ -114,12 +114,13 @@ def next_image(embs, ys):
 
 
 def start(_, embs, ys):
-    embs, ys = next_image(embs, ys)
+    image, 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),
+            image
             embs,
             ys,
             ]
@@ -130,11 +131,11 @@ def choose(choice, embs, ys):
         choice = 1
     elif choice == 'Neither':
         _ = embs.pop(-1)
-        embs, ys = next_image(embs, ys)
+        img, embs, ys = next_image(embs, ys)
     else:
         choice = 0
     ys.append(choice)
-    embs, ys = next_image(embs, ys)
+    img, 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;}"