rynmurdock commited on
Commit
ae7b7f1
1 Parent(s): 8b05017

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -129,6 +129,7 @@ def choose(choice, embs, ys, calibrate_prompts):
129
  elif choice == 'Neither':
130
  _ = embs.pop(-1)
131
  img, embs, ys, calibrate_prompts = next_image(embs, ys, calibrate_prompts)
 
132
  else:
133
  choice = 0
134
  ys.append(choice)
 
129
  elif choice == 'Neither':
130
  _ = embs.pop(-1)
131
  img, embs, ys, calibrate_prompts = next_image(embs, ys, calibrate_prompts)
132
+ return img, embs, ys, calibrate_prompts
133
  else:
134
  choice = 0
135
  ys.append(choice)