Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -26,8 +26,8 @@ def inference_caption(image):
|
|
| 26 |
im,
|
| 27 |
generation_type="beam_search",
|
| 28 |
top_p=1.0,
|
| 29 |
-
min_seq_len=
|
| 30 |
-
seq_len=
|
| 31 |
repetition_penalty=1.2
|
| 32 |
)
|
| 33 |
return open_clip.decode(generated[0].detach()).split("<end_of_text>")[0].replace("<start_of_text>", "")
|
|
|
|
| 26 |
im,
|
| 27 |
generation_type="beam_search",
|
| 28 |
top_p=1.0,
|
| 29 |
+
min_seq_len=5,
|
| 30 |
+
seq_len=100,
|
| 31 |
repetition_penalty=1.2
|
| 32 |
)
|
| 33 |
return open_clip.decode(generated[0].detach()).split("<end_of_text>")[0].replace("<start_of_text>", "")
|