Update app.py

app.py

@@ -30,6 +30,14 @@ glob_idx = 0
 def next_image(embs, ys, calibrate_prompts):
     global glob_idx
     glob_idx = glob_idx + 1
+
+    # handle case where every instance of calibration prompts is 'Neither' or 'Like' or 'Dislike'
+    if len(calibrate_prompts) == 0 and len(list(set(ys))) <= 1:
+        embs.append(torch.zeros(1, 1280))
+        embs.append(torch.zeros(1, 1280))
+        ys.append(0)
+        ys.append(1)
+        
     with torch.no_grad():
         if len(calibrate_prompts) > 0:
             print('######### Calibrating with sample prompts #########')