prithivMLmods commited on
Commit
712f536
·
verified ·
1 Parent(s): 101b503

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -18,7 +18,6 @@ import requests
18
  from transformers import (
19
  Qwen2VLForConditionalGeneration,
20
  Qwen2_5_VLForConditionalGeneration,
21
- Qwen3VLForConditionalGeneration,
22
  AutoModelForCausalLM,
23
  AutoModelForVision2Seq,
24
  AutoProcessor,
@@ -115,9 +114,9 @@ MAX_INPUT_TOKEN_LENGTH = int(os.getenv("MAX_INPUT_TOKEN_LENGTH", "4096"))
115
  device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
116
 
117
  # Load Nanonets-OCR-s
118
- MODEL_ID_M = "prithivMLmods/Qwen3VL-Hybrid-2507-5B"
119
  processor_m = AutoProcessor.from_pretrained(MODEL_ID_M, trust_remote_code=True)
120
- model_m = Qwen3VLForConditionalGeneration.from_pretrained(
121
  MODEL_ID_M,
122
  trust_remote_code=True,
123
  torch_dtype=torch.float16
 
18
  from transformers import (
19
  Qwen2VLForConditionalGeneration,
20
  Qwen2_5_VLForConditionalGeneration,
 
21
  AutoModelForCausalLM,
22
  AutoModelForVision2Seq,
23
  AutoProcessor,
 
114
  device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
115
 
116
  # Load Nanonets-OCR-s
117
+ MODEL_ID_M = "nanonets/Nanonets-OCR-s"
118
  processor_m = AutoProcessor.from_pretrained(MODEL_ID_M, trust_remote_code=True)
119
+ model_m = Qwen2_5_VLForConditionalGeneration.from_pretrained(
120
  MODEL_ID_M,
121
  trust_remote_code=True,
122
  torch_dtype=torch.float16