rynmurdock commited on
Commit
8b05017
1 Parent(s): 2990438

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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.zeros(1, 1280))
37
- embs.append(torch.zeros(1, 1280))
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