Spaces:
Running
on
A10G
Running
on
A10G
rynmurdock
commited on
Commit
•
8b05017
1
Parent(s):
2990438
Update app.py
Browse files
app.py
CHANGED
@@ -33,8 +33,8 @@ def next_image(embs, ys, calibrate_prompts):
|
|
33 |
|
34 |
# handle case where every instance of calibration prompts is 'Neither' or 'Like' or 'Dislike'
|
35 |
if len(calibrate_prompts) == 0 and len(list(set(ys))) <= 1:
|
36 |
-
embs.append(torch.
|
37 |
-
embs.append(torch.
|
38 |
ys.append(0)
|
39 |
ys.append(1)
|
40 |
|
|
|
33 |
|
34 |
# handle case where every instance of calibration prompts is 'Neither' or 'Like' or 'Dislike'
|
35 |
if len(calibrate_prompts) == 0 and len(list(set(ys))) <= 1:
|
36 |
+
embs.append(.01*torch.randn(1, 1280))
|
37 |
+
embs.append(.01*torch.randn(1, 1280))
|
38 |
ys.append(0)
|
39 |
ys.append(1)
|
40 |
|