Spaces:
Build error
Build error
Update app.py
Browse files
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 = "
|
| 119 |
processor_m = AutoProcessor.from_pretrained(MODEL_ID_M, trust_remote_code=True)
|
| 120 |
-
model_m =
|
| 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
|