rynmurdock commited on
Commit
225ad9c
1 Parent(s): b41bb43
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -210,8 +210,8 @@ def next_image(embs, img_embs, ys, calibrate_prompts):
210
 
211
  # handle case where every instance of calibration prompts is 'Neither' or 'Like' or 'Dislike'
212
  if len(calibrate_prompts) == 0 and len(list(set(ys))) <= 1:
213
- embs.append(.01*torch.randn(1, 1024))
214
- embs.append(.01*torch.randn(1, 1024))
215
  img_embs.append(.01*torch.randn(1, 1024))
216
  img_embs.append(.01*torch.randn(1, 1024))
217
  ys.append(0)
@@ -341,7 +341,7 @@ document.body.addEventListener('click', function(event) {
341
 
342
  with gr.Blocks(css=css, head=js_head) as demo:
343
  gr.Markdown('''### Zahir: Generative Recommenders for Unprompted, Scalable Exploration
344
- Explore the latent space without prompting based on your feedback. Learn more on [the write-up](https://rynmurdock.github.io/posts/2024/3/generative_recomenders/).
345
  ''', elem_id="description")
346
  embs = gr.State([])
347
  img_embs = gr.State([])
 
210
 
211
  # handle case where every instance of calibration prompts is 'Neither' or 'Like' or 'Dislike'
212
  if len(calibrate_prompts) == 0 and len(list(set(ys))) <= 1:
213
+ embs.append(.01*torch.randn(1, 2048))
214
+ embs.append(.01*torch.randn(1, 2048))
215
  img_embs.append(.01*torch.randn(1, 1024))
216
  img_embs.append(.01*torch.randn(1, 1024))
217
  ys.append(0)
 
341
 
342
  with gr.Blocks(css=css, head=js_head) as demo:
343
  gr.Markdown('''### Zahir: Generative Recommenders for Unprompted, Scalable Exploration
344
+ Explore the latent space without prompting based on your feedbackg. Learn more on [the write-up](https://rynmurdock.github.io/posts/2024/3/generative_recomenders/).
345
  ''', elem_id="description")
346
  embs = gr.State([])
347
  img_embs = gr.State([])