Update app.py
Browse files
app.py
CHANGED
@@ -15,7 +15,7 @@ def input_changes(input_img):
|
|
15 |
if input_img == None:
|
16 |
return manual_cap.update(value="",placeholder=ph_message), caption_output.update(value=None), sound_output.update(value=None)
|
17 |
else:
|
18 |
-
cap = caption(input_img, fn_index=0)
|
19 |
print("CoCa caption: '" + cap + "' β’ ")
|
20 |
ph_update = "CoCa caption: '" + cap + "' β’ "
|
21 |
|
|
|
15 |
if input_img == None:
|
16 |
return manual_cap.update(value="",placeholder=ph_message), caption_output.update(value=None), sound_output.update(value=None)
|
17 |
else:
|
18 |
+
cap = caption(input_img, decoding_method="Beam search", rep_penalty=1.2, top_p=0.5, min_seq_len=5, seq_len=20, fn_index=0)
|
19 |
print("CoCa caption: '" + cap + "' β’ ")
|
20 |
ph_update = "CoCa caption: '" + cap + "' β’ "
|
21 |
|