prithivMLmods commited on
Commit
5780205
·
verified ·
1 Parent(s): 28a6d3f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -6
app.py CHANGED
@@ -133,16 +133,13 @@ model_v = Qwen2_5_VLForConditionalGeneration.from_pretrained(
133
  torch_dtype=torch.float16
134
  ).to(device).eval()
135
 
136
- # Load PaddleOCR-VL
137
- MODEL_ID_P = "PaddlePaddle/PaddleOCR-VL"
138
- SUBFOLDER_P = "PaddleOCR-VL-0.9B"
139
- processor_p = AutoProcessor.from_pretrained(MODEL_ID_P, trust_remote_code=True, subfolder=SUBFOLDER_P)
140
  model_p = AutoModelForCausalLM.from_pretrained(
141
  MODEL_ID_P,
142
  trust_remote_code=True,
143
- subfolder=SUBFOLDER_P,
144
  torch_dtype=torch.float16,
145
- #attn_implementation="flash_attention_2"
146
  ).to(device).eval()
147
 
148
 
 
133
  torch_dtype=torch.float16
134
  ).to(device).eval()
135
 
136
+ # Load PaddleOCR-VL [PaddlePaddle/PaddleOCR-VL]
137
+ MODEL_ID_P = "strangervisionhf/paddle"
138
+ processor_p = AutoProcessor.from_pretrained(MODEL_ID_P, trust_remote_code=True)
 
139
  model_p = AutoModelForCausalLM.from_pretrained(
140
  MODEL_ID_P,
141
  trust_remote_code=True,
 
142
  torch_dtype=torch.float16,
 
143
  ).to(device).eval()
144
 
145