rynmurdock commited on
Commit
93b9a94
1 Parent(s): 6f68207

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -139,7 +139,7 @@ def next_image(embs, ys, calibrate_prompts):
139
  rng_prompt = random.choice(prompt_list)
140
  w = 1# if len(embs) % 2 == 0 else 0
141
  im_emb = w * lin_class.coef_.to(device=DEVICE, dtype=torch.float16)
142
- prompt= '' if glob_idx % 2 == 0 else rng_prompt
143
  print(prompt, len(ys))
144
  image, im_emb = predict(prompt, im_emb)
145
  embs.append(im_emb)
 
139
  rng_prompt = random.choice(prompt_list)
140
  w = 1# if len(embs) % 2 == 0 else 0
141
  im_emb = w * lin_class.coef_.to(device=DEVICE, dtype=torch.float16)
142
+ prompt= 'an image' if glob_idx % 2 == 0 else rng_prompt
143
  print(prompt, len(ys))
144
  image, im_emb = predict(prompt, im_emb)
145
  embs.append(im_emb)