Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -61,7 +61,8 @@ def run_example(task_prompt, image, text_input=None, model_id='microsoft/Florenc
|
|
| 61 |
max_new_tokens=1024,
|
| 62 |
early_stopping=False,
|
| 63 |
do_sample=False,
|
| 64 |
-
num_beams=
|
|
|
|
| 65 |
)
|
| 66 |
generated_text = processor.batch_decode(generated_ids, skip_special_tokens=False)[0]
|
| 67 |
parsed_answer = processor.post_process_generation(
|
|
|
|
| 61 |
max_new_tokens=1024,
|
| 62 |
early_stopping=False,
|
| 63 |
do_sample=False,
|
| 64 |
+
num_beams=1,
|
| 65 |
+
use_cache=False,
|
| 66 |
)
|
| 67 |
generated_text = processor.batch_decode(generated_ids, skip_special_tokens=False)[0]
|
| 68 |
parsed_answer = processor.post_process_generation(
|