Spaces:
Running on Zero
Running on Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -140,7 +140,7 @@ model_m = Qwen2_5_VLForConditionalGeneration.from_pretrained(
|
|
| 140 |
).to(device).eval()
|
| 141 |
|
| 142 |
# Load Nanonets-OCR2-1.5B-exp
|
| 143 |
-
MODEL_ID_N = "strangervisionhf/excess_layer_pruned-nanonets-1.5b"
|
| 144 |
processor_n = AutoProcessor.from_pretrained(MODEL_ID_N, trust_remote_code=True)
|
| 145 |
model_n = AutoModelForImageTextToText.from_pretrained(
|
| 146 |
MODEL_ID_N,
|
|
@@ -162,7 +162,7 @@ model_d = AutoModelForCausalLM.from_pretrained(
|
|
| 162 |
).eval()
|
| 163 |
|
| 164 |
# Load PaddleOCR
|
| 165 |
-
MODEL_ID_P = "strangervisionhf/paddle"
|
| 166 |
processor_p = AutoProcessor.from_pretrained(MODEL_ID_P, trust_remote_code=True)
|
| 167 |
model_p = AutoModelForCausalLM.from_pretrained(
|
| 168 |
MODEL_ID_P,
|
|
|
|
| 140 |
).to(device).eval()
|
| 141 |
|
| 142 |
# Load Nanonets-OCR2-1.5B-exp
|
| 143 |
+
MODEL_ID_N = "strangervisionhf/excess_layer_pruned-nanonets-1.5b" # -> https://huggingface.co/nanonets/Nanonets-OCR2-1.5B-exp
|
| 144 |
processor_n = AutoProcessor.from_pretrained(MODEL_ID_N, trust_remote_code=True)
|
| 145 |
model_n = AutoModelForImageTextToText.from_pretrained(
|
| 146 |
MODEL_ID_N,
|
|
|
|
| 162 |
).eval()
|
| 163 |
|
| 164 |
# Load PaddleOCR
|
| 165 |
+
MODEL_ID_P = "strangervisionhf/paddle" # -> https://huggingface.co/PaddlePaddle/PaddleOCR-VL
|
| 166 |
processor_p = AutoProcessor.from_pretrained(MODEL_ID_P, trust_remote_code=True)
|
| 167 |
model_p = AutoModelForCausalLM.from_pretrained(
|
| 168 |
MODEL_ID_P,
|