prithivMLmods commited on
Commit
47f6cdd
·
verified ·
1 Parent(s): 917081d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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,