Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -264,8 +264,8 @@ def captioning(img):
|
|
264 |
**inputsa,
|
265 |
do_sample=False,
|
266 |
num_beams=5,
|
267 |
-
max_length=
|
268 |
-
min_length=
|
269 |
top_p=0.9,
|
270 |
repetition_penalty=1.5,
|
271 |
length_penalty=1.0,
|
@@ -284,8 +284,8 @@ def captioning(img):
|
|
284 |
**inputs,
|
285 |
do_sample=False,
|
286 |
num_beams=5,
|
287 |
-
max_length=
|
288 |
-
min_length=
|
289 |
top_p=0.9,
|
290 |
repetition_penalty=1.5,
|
291 |
length_penalty=1.0,
|
@@ -304,8 +304,8 @@ def captioning(img):
|
|
304 |
**inputf,
|
305 |
do_sample=False,
|
306 |
num_beams=5,
|
307 |
-
max_length=
|
308 |
-
min_length=
|
309 |
top_p=0.9,
|
310 |
repetition_penalty=1.5,
|
311 |
length_penalty=1.0,
|
|
|
264 |
**inputsa,
|
265 |
do_sample=False,
|
266 |
num_beams=5,
|
267 |
+
max_length=768,
|
268 |
+
min_length=256,
|
269 |
top_p=0.9,
|
270 |
repetition_penalty=1.5,
|
271 |
length_penalty=1.0,
|
|
|
284 |
**inputs,
|
285 |
do_sample=False,
|
286 |
num_beams=5,
|
287 |
+
max_length=96,
|
288 |
+
min_length=16,
|
289 |
top_p=0.9,
|
290 |
repetition_penalty=1.5,
|
291 |
length_penalty=1.0,
|
|
|
304 |
**inputf,
|
305 |
do_sample=False,
|
306 |
num_beams=5,
|
307 |
+
max_length=192,
|
308 |
+
min_length=96,
|
309 |
top_p=0.9,
|
310 |
repetition_penalty=1.5,
|
311 |
length_penalty=1.0,
|