Update app.py

app.py

@@ -121,7 +121,7 @@ def next_image(embs, ys, calibrate_prompts):
             im_emb = w * lin_class.coef_.to(device=DEVICE, dtype=torch.float16)
             prompt= 'an image' if glob_idx % 2 == 0 else rng_prompt
             print(prompt)
-            image, im_emb = predict(prompt, img_emb)
+            image, im_emb = predict(prompt, im_emb)
             embs.append(im_emb)
             return image, embs, ys, calibrate_prompts