Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -55,7 +55,7 @@ def process_image_and_text(image_pil, num_beams, temperature, seed):
|
|
55 |
torch.manual_seed(seed) # Set random seed for reproducibility
|
56 |
|
57 |
with torch.inference_mode():
|
58 |
-
generation = model.generate(**model_inputs, max_new_tokens=
|
59 |
generation = generation[0][input_len:]
|
60 |
decoded = processor.decode(generation, skip_special_tokens=True)
|
61 |
|
|
|
55 |
torch.manual_seed(seed) # Set random seed for reproducibility
|
56 |
|
57 |
with torch.inference_mode():
|
58 |
+
generation = model.generate(**model_inputs, max_new_tokens=200, do_sample=False, num_beams=num_beams, temperature=temperature)
|
59 |
generation = generation[0][input_len:]
|
60 |
decoded = processor.decode(generation, skip_special_tokens=True)
|
61 |
|