Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
@@ -22,8 +22,7 @@ def predict(image: Image.Image, question: str) -> str:
|
|
22 |
generated_ids = model.generate(
|
23 |
pixel_values=pixel_values,
|
24 |
input_ids=input_ids,
|
25 |
-
max_new_tokens=50
|
26 |
-
num_beams=5
|
27 |
)
|
28 |
|
29 |
answer = model.text_processor.batch_decode(generated_ids, skip_special_tokens=True)[0].strip()
|
|
|
22 |
generated_ids = model.generate(
|
23 |
pixel_values=pixel_values,
|
24 |
input_ids=input_ids,
|
25 |
+
max_new_tokens=50
|
|
|
26 |
)
|
27 |
|
28 |
answer = model.text_processor.batch_decode(generated_ids, skip_special_tokens=True)[0].strip()
|