Update model.py
Browse files
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, "
|
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 |
|