Image-Caption-2-Shap-E

Paused

fffiloni commited on Oct 4, 2023

Commit

40577ce

1 Parent(s): 66878be

Update model.py

Files changed (1) hide show

model.py CHANGED Viewed

@@ -17,7 +17,7 @@ from shap_e.util.image_util import load_image
 caption = gr.load(name="spaces/fffiloni/CoCa-clone")
 def create_image_caption(image_init):
-    cap = caption(image_init, "Beam search", 1.2, 0.5, 5, 20, fn_index=0)
     print("cap: " + cap)
     return cap

 caption = gr.load(name="spaces/fffiloni/CoCa-clone")
 def create_image_caption(image_init):
+    cap = caption(image_init, "Nucleus sampling", 1.2, 0.5, 5, 20, fn_index=0)
     print("cap: " + cap)
     return cap