Update app.py

app.py

@@ -101,8 +101,8 @@ def next_image(embs, ys, calibrate_prompts):
 
 
 
-def start(_, embs, ys):
-    image, embs, ys, calibrate_prompts = next_image(embs, ys)
+def start(_, embs, ys, calibrate_prompts):
+    image, embs, ys, calibrate_prompts = next_image(embs, ys, calibrate_prompts)
     return [
             gr.Button(value='Like', interactive=True), 
             gr.Button(value='Neither', interactive=True), 
@@ -115,16 +115,16 @@ def start(_, embs, ys):
             ]
 
 
-def choose(choice, embs, ys):
+def choose(choice, embs, ys, calibrate_prompts):
     if choice == 'Like':
         choice = 1
     elif choice == 'Neither':
         _ = embs.pop(-1)
-        img, embs, ys, calibrate_prompts = next_image(embs, ys)
+        img, embs, ys, calibrate_prompts = next_image(embs, ys, calibrate_prompts)
     else:
         choice = 0
     ys.append(choice)
-    img, embs, ys, calibrate_prompts = next_image(embs, ys)
+    img, embs, ys, calibrate_prompts = next_image(embs, ys, calibrate_prompts)
     return img, embs, ys, calibrate_prompts
 
 css = "div#output-image {height: 768px !important; width: 768px !important; margin:auto;}"