rynmurdock commited on
Commit
19701ae
1 Parent(s): 0186388

neither fix

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -277,6 +277,7 @@ def choose(choice, embs, img_embs, ys, calibrate_prompts):
277
  choice = 1
278
  elif choice == 'Neither (Space)':
279
  _ = embs.pop(-1)
 
280
  img, embs, img_embs, ys, calibrate_prompts = next_image(embs, img_embs, ys, calibrate_prompts)
281
  return img, embs, img_embs, ys, calibrate_prompts
282
  else:
 
277
  choice = 1
278
  elif choice == 'Neither (Space)':
279
  _ = embs.pop(-1)
280
+ _ = img_embs.pop(-1)
281
  img, embs, img_embs, ys, calibrate_prompts = next_image(embs, img_embs, ys, calibrate_prompts)
282
  return img, embs, img_embs, ys, calibrate_prompts
283
  else: