Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
275bb85
1
Parent(s):
e1b1045
Increase max_new_tokens in HF_PIPE for enhanced prediction capacity
Browse files
app.py
CHANGED
|
@@ -89,7 +89,7 @@ def predict(pil_image):
|
|
| 89 |
]
|
| 90 |
|
| 91 |
# Use the pipeline with the properly formatted messages
|
| 92 |
-
return HF_PIPE(messages)
|
| 93 |
|
| 94 |
def run_hf_ocr(image_path):
|
| 95 |
"""
|
|
|
|
| 89 |
]
|
| 90 |
|
| 91 |
# Use the pipeline with the properly formatted messages
|
| 92 |
+
return HF_PIPE(messages,max_new_tokens=8096)
|
| 93 |
|
| 94 |
def run_hf_ocr(image_path):
|
| 95 |
"""
|