Spaces:
Sleeping
Sleeping
Luis J Camargo
commited on
Commit
·
919ae04
1
Parent(s):
bef2fad
temperature test
Browse files
app.py
CHANGED
|
@@ -88,7 +88,7 @@ class PaddleOCRModelManager(object):
|
|
| 88 |
def download_model():
|
| 89 |
"""Download the fine-tuned Tachiwin model from Hugging Face"""
|
| 90 |
model_repo = "tachiwin/PaddleOCR-VL-Tachiwin-BF16"
|
| 91 |
-
model_repo = "unsloth/PaddleOCR-VL"
|
| 92 |
model_dir = "./tachiwin_model"
|
| 93 |
|
| 94 |
print(f"Downloading Tachiwin model from {model_repo}...")
|
|
@@ -134,7 +134,7 @@ def inference(img):
|
|
| 134 |
return "Please upload an image."
|
| 135 |
|
| 136 |
try:
|
| 137 |
-
return model_manager.infer(img)
|
| 138 |
|
| 139 |
# # Now extract text from the serialized structure
|
| 140 |
# extracted_texts = []
|
|
|
|
| 88 |
def download_model():
|
| 89 |
"""Download the fine-tuned Tachiwin model from Hugging Face"""
|
| 90 |
model_repo = "tachiwin/PaddleOCR-VL-Tachiwin-BF16"
|
| 91 |
+
#model_repo = "unsloth/PaddleOCR-VL"
|
| 92 |
model_dir = "./tachiwin_model"
|
| 93 |
|
| 94 |
print(f"Downloading Tachiwin model from {model_repo}...")
|
|
|
|
| 134 |
return "Please upload an image."
|
| 135 |
|
| 136 |
try:
|
| 137 |
+
return model_manager.infer(img, min_new_tokens=3, temperature=1.5, min_p=0.1)
|
| 138 |
|
| 139 |
# # Now extract text from the serialized structure
|
| 140 |
# extracted_texts = []
|