Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -263,7 +263,7 @@ def captioning(img):
|
|
263 |
'''
|
264 |
generated_ids = model5.generate(
|
265 |
**inputsa,
|
266 |
-
do_sample=
|
267 |
num_beams=5,
|
268 |
max_length=128,
|
269 |
min_length=64,
|
@@ -275,7 +275,7 @@ def captioning(img):
|
|
275 |
'''
|
276 |
generated_ids = model5.generate(
|
277 |
**inputsa,
|
278 |
-
do_sample=
|
279 |
num_beams=1,
|
280 |
max_length=128,
|
281 |
min_length=64,
|
@@ -296,7 +296,7 @@ def captioning(img):
|
|
296 |
#with torch.no_grad():
|
297 |
generated_ids = model5.generate(
|
298 |
**inputs,
|
299 |
-
do_sample=
|
300 |
num_beams=1,
|
301 |
max_length=64,
|
302 |
#min_length=16,
|
@@ -316,7 +316,7 @@ def captioning(img):
|
|
316 |
).to('cuda')
|
317 |
generated_ids = model5.generate(
|
318 |
**inputf,
|
319 |
-
do_sample=
|
320 |
num_beams=1,
|
321 |
max_length=96,
|
322 |
min_length=64,
|
|
|
263 |
'''
|
264 |
generated_ids = model5.generate(
|
265 |
**inputsa,
|
266 |
+
do_sample=True,
|
267 |
num_beams=5,
|
268 |
max_length=128,
|
269 |
min_length=64,
|
|
|
275 |
'''
|
276 |
generated_ids = model5.generate(
|
277 |
**inputsa,
|
278 |
+
do_sample=True,
|
279 |
num_beams=1,
|
280 |
max_length=128,
|
281 |
min_length=64,
|
|
|
296 |
#with torch.no_grad():
|
297 |
generated_ids = model5.generate(
|
298 |
**inputs,
|
299 |
+
do_sample=True,
|
300 |
num_beams=1,
|
301 |
max_length=64,
|
302 |
#min_length=16,
|
|
|
316 |
).to('cuda')
|
317 |
generated_ids = model5.generate(
|
318 |
**inputf,
|
319 |
+
do_sample=True,
|
320 |
num_beams=1,
|
321 |
max_length=96,
|
322 |
min_length=64,
|