breadlicker45 commited on
Commit
dbffcc6
·
verified ·
1 Parent(s): a1117d5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -52,7 +52,7 @@ def process_image_and_text(image_pil, text_input, num_beams, temperature, seed):
52
  torch.manual_seed(seed) # Set random seed for reproducibility
53
 
54
  with torch.inference_mode():
55
- generation = model.generate(**model_inputs, max_new_tokens=100, do_sample=True, num_beams=num_beams, temperature=temperature)
56
  generation = generation[0][input_len:]
57
  decoded = processor.decode(generation, skip_special_tokens=True)
58
 
 
52
  torch.manual_seed(seed) # Set random seed for reproducibility
53
 
54
  with torch.inference_mode():
55
+ generation = model.generate(**model_inputs, max_new_tokens=500, do_sample=True, num_beams=num_beams, temperature=temperature)
56
  generation = generation[0][input_len:]
57
  decoded = processor.decode(generation, skip_special_tokens=True)
58