prithivMLmods commited on
Commit
238ed44
·
verified ·
1 Parent(s): 0e52c4c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -5
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 = "PaddlePaddle/PaddleOCR-VL"
137
- SUBFOLDER_P = "PaddleOCR-VL-0.9B"
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