rynmurdock commited on
Commit
23702ab
1 Parent(s): db551d5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -117,7 +117,7 @@ def next_image(embs, ys, calibrate_prompts):
117
 
118
  w = 1# if len(embs) % 2 == 0 else 0
119
  im_emb = w * lin_class.coef_.to(device=DEVICE, dtype=torch.float16)
120
- prompt= 'an image' if glob_idx % 2 == 0 else rng_prompt
121
  print(prompt)
122
  image, im_emb = predict(prompt, im_emb)
123
  embs.append(im_emb)
 
117
 
118
  w = 1# if len(embs) % 2 == 0 else 0
119
  im_emb = w * lin_class.coef_.to(device=DEVICE, dtype=torch.float16)
120
+ prompt= '' if glob_idx % 2 == 0 else rng_prompt
121
  print(prompt)
122
  image, im_emb = predict(prompt, im_emb)
123
  embs.append(im_emb)