Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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 =
|
| 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
|