fffiloni commited on
Commit
40577ce
1 Parent(s): 66878be

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +1 -1
model.py CHANGED
@@ -17,7 +17,7 @@ from shap_e.util.image_util import load_image
17
  caption = gr.load(name="spaces/fffiloni/CoCa-clone")
18
 
19
  def create_image_caption(image_init):
20
- cap = caption(image_init, "Beam search", 1.2, 0.5, 5, 20, fn_index=0)
21
  print("cap: " + cap)
22
  return cap
23
 
 
17
  caption = gr.load(name="spaces/fffiloni/CoCa-clone")
18
 
19
  def create_image_caption(image_init):
20
+ cap = caption(image_init, "Nucleus sampling", 1.2, 0.5, 5, 20, fn_index=0)
21
  print("cap: " + cap)
22
  return cap
23