Update app.py
Browse files
app.py
CHANGED
|
@@ -73,7 +73,7 @@ qwen_model = Qwen2VLForConditionalGeneration.from_pretrained(
|
|
| 73 |
).to("cuda").eval()
|
| 74 |
|
| 75 |
# RolmOCR branch (@RolmOCR)
|
| 76 |
-
ROLMOCR_MODEL_ID = "
|
| 77 |
rolmocr_processor = AutoProcessor.from_pretrained(ROLMOCR_MODEL_ID, trust_remote_code=True)
|
| 78 |
rolmocr_model = Qwen2_5_VLForConditionalGeneration.from_pretrained(
|
| 79 |
ROLMOCR_MODEL_ID,
|
|
|
|
| 73 |
).to("cuda").eval()
|
| 74 |
|
| 75 |
# RolmOCR branch (@RolmOCR)
|
| 76 |
+
ROLMOCR_MODEL_ID = "reducto/RolmOCR"
|
| 77 |
rolmocr_processor = AutoProcessor.from_pretrained(ROLMOCR_MODEL_ID, trust_remote_code=True)
|
| 78 |
rolmocr_model = Qwen2_5_VLForConditionalGeneration.from_pretrained(
|
| 79 |
ROLMOCR_MODEL_ID,
|