Update app.py
Browse filestokenization update
app.py
CHANGED
|
@@ -96,7 +96,8 @@ def solve(req: SolveRequest, _: str = Depends(verify_key)):
|
|
| 96 |
|
| 97 |
generated_ids = model.generate(
|
| 98 |
pixel_values,
|
| 99 |
-
max_new_tokens=
|
|
|
|
| 100 |
num_beams=4,
|
| 101 |
do_sample=False
|
| 102 |
)
|
|
|
|
| 96 |
|
| 97 |
generated_ids = model.generate(
|
| 98 |
pixel_values,
|
| 99 |
+
max_new_tokens=5,
|
| 100 |
+
min_new_tokens=5,
|
| 101 |
num_beams=4,
|
| 102 |
do_sample=False
|
| 103 |
)
|