1inkusFace commited on
Commit
b0ca3cf
·
verified ·
1 Parent(s): 314d6d0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -263,7 +263,7 @@ def captioning(img):
263
  '''
264
  generated_ids = model5.generate(
265
  **inputsa,
266
- do_sample=False,
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=False,
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=False,
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=False,
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,