Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -132,14 +132,12 @@ model_v = Qwen2_5_VLForConditionalGeneration.from_pretrained(
|
|
| 132 |
torch_dtype=torch.float16
|
| 133 |
).to(device).eval()
|
| 134 |
|
| 135 |
-
# Load PaddleOCR-VL
|
| 136 |
-
MODEL_ID_P = "
|
| 137 |
-
|
| 138 |
-
processor_p = AutoProcessor.from_pretrained(MODEL_ID_P, trust_remote_code=True, subfolder=SUBFOLDER_P)
|
| 139 |
model_p = AutoModelForCausalLM.from_pretrained(
|
| 140 |
MODEL_ID_P,
|
| 141 |
trust_remote_code=True,
|
| 142 |
-
subfolder=SUBFOLDER_P,
|
| 143 |
torch_dtype=torch.float16
|
| 144 |
).to(device).eval()
|
| 145 |
|
|
|
|
| 132 |
torch_dtype=torch.float16
|
| 133 |
).to(device).eval()
|
| 134 |
|
| 135 |
+
# Load PaddleOCR-VL [PaddlePaddle/PaddleOCR-VL]
|
| 136 |
+
MODEL_ID_P = "strangervisionhf/paddle"
|
| 137 |
+
processor_p = AutoProcessor.from_pretrained(MODEL_ID_P, trust_remote_code=True)
|
|
|
|
| 138 |
model_p = AutoModelForCausalLM.from_pretrained(
|
| 139 |
MODEL_ID_P,
|
| 140 |
trust_remote_code=True,
|
|
|
|
| 141 |
torch_dtype=torch.float16
|
| 142 |
).to(device).eval()
|
| 143 |
|