Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -328,9 +328,9 @@ def captioning(img, prompt_1,prompt_2,prompt_3,prompt_4,prompt_5):
|
|
328 |
**inputs,
|
329 |
do_sample=True,
|
330 |
num_beams=5,
|
331 |
-
max_length=
|
332 |
-
|
333 |
-
top_p=0.
|
334 |
repetition_penalty=1.0,
|
335 |
length_penalty=2.0,
|
336 |
temperature=0.55,
|
|
|
328 |
**inputs,
|
329 |
do_sample=True,
|
330 |
num_beams=5,
|
331 |
+
max_length=256,
|
332 |
+
min_length=42,
|
333 |
+
top_p=0.2,
|
334 |
repetition_penalty=1.0,
|
335 |
length_penalty=2.0,
|
336 |
temperature=0.55,
|