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