prithivMLmods commited on
Commit
36bcca3
·
verified ·
1 Parent(s): d91e909

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -18,6 +18,7 @@ import requests
18
  from transformers import (
19
  Qwen2VLForConditionalGeneration,
20
  Qwen2_5_VLForConditionalGeneration,
 
21
  AutoModelForCausalLM,
22
  AutoModelForVision2Seq,
23
  AutoProcessor,
@@ -116,7 +117,7 @@ device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
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
 
18
  from transformers import (
19
  Qwen2VLForConditionalGeneration,
20
  Qwen2_5_VLForConditionalGeneration,
21
+ Qwen3VLForConditionalGeneration,
22
  AutoModelForCausalLM,
23
  AutoModelForVision2Seq,
24
  AutoProcessor,
 
117
  # Load Nanonets-OCR-s
118
  MODEL_ID_M = "nanonets/Nanonets-OCR-s"
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