Spaces:
Running
on
A10G
Running
on
A10G
rynmurdock
commited on
Commit
•
93b9a94
1
Parent(s):
6f68207
Update app.py
Browse files
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)
|