prithivMLmods commited on
Commit
d7c41ec
·
verified ·
1 Parent(s): 49e561b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -144,7 +144,7 @@ model_m = Qwen2_5_VLForConditionalGeneration.from_pretrained(
144
  ).to(device).eval()
145
 
146
  # Load Nanonets-OCR2-1.5B-exp
147
- MODEL_ID_N = "nanonets/Nanonets-OCR2-1.5B-exp"
148
  processor_n = AutoProcessor.from_pretrained(MODEL_ID_N, trust_remote_code=True)
149
  model_n = Qwen2_5_VLForConditionalGeneration.from_pretrained(
150
  MODEL_ID_N,
 
144
  ).to(device).eval()
145
 
146
  # Load Nanonets-OCR2-1.5B-exp
147
+ MODEL_ID_N = "strangervisionhf/excess_layer_pruned-nanonets-1.5b"
148
  processor_n = AutoProcessor.from_pretrained(MODEL_ID_N, trust_remote_code=True)
149
  model_n = Qwen2_5_VLForConditionalGeneration.from_pretrained(
150
  MODEL_ID_N,