Update app.py

app.py

@@ -44,7 +44,7 @@ def next_image(embs, ys, calibrate_prompts):
             image = Image.open(BytesIO(response.content))
 
             embs.append(torch.tensor([float(i) for i in urlopen(output['file2']).read().decode('utf-8').split(', ')]).unsqueeze(0))
-            return image, embs, ys
+            return image, embs, ys, calibrate_prompts
         else:
             print('######### Roaming #########')