Spaces:

rynmurdock
/

generative_recsys

Running on A10G

rynmurdock commited on Apr 11

Commit

2bc76ca

•

1 Parent(s): c8ea988

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -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)
             embs.append(im_emb)
             return 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, im_emb)
             embs.append(im_emb)
             return image, embs, ys, calibrate_prompts