Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -91,8 +91,8 @@ model_i = Qwen2_5_VLForConditionalGeneration.from_pretrained(
|
|
91 |
|
92 |
#-----------------------------------------#
|
93 |
|
94 |
-
# Load
|
95 |
-
MODEL_ID_J = "
|
96 |
processor_j = AutoProcessor.from_pretrained(
|
97 |
MODEL_ID_J,
|
98 |
trust_remote_code=True
|
@@ -154,7 +154,7 @@ def process_document_stream(model_name: str, task_choice: str, image: Image.Imag
|
|
154 |
elif model_name == "Nanonets-OCR-s": processor, model = processor_c, model_c
|
155 |
elif model_name == "MonkeyOCR-Recognition": processor, model = processor_g, model_g
|
156 |
elif model_name == "olmOCR-7B-0725": processor, model = processor_i, model_i
|
157 |
-
elif model_name == "
|
158 |
else:
|
159 |
yield "Invalid model selected.", "Invalid model selected.", None
|
160 |
return
|
@@ -224,7 +224,7 @@ def create_gradio_interface():
|
|
224 |
"olmOCR-7B-0725",
|
225 |
"Nanonets-OCR-s",
|
226 |
"Megalodon-OCR-Sync-0713",
|
227 |
-
"
|
228 |
],
|
229 |
label="Select Model", value="Nanonets-OCR-s"
|
230 |
)
|
|
|
91 |
|
92 |
#-----------------------------------------#
|
93 |
|
94 |
+
# Load RolmOCR
|
95 |
+
MODEL_ID_J = "reducto/RolmOCR"
|
96 |
processor_j = AutoProcessor.from_pretrained(
|
97 |
MODEL_ID_J,
|
98 |
trust_remote_code=True
|
|
|
154 |
elif model_name == "Nanonets-OCR-s": processor, model = processor_c, model_c
|
155 |
elif model_name == "MonkeyOCR-Recognition": processor, model = processor_g, model_g
|
156 |
elif model_name == "olmOCR-7B-0725": processor, model = processor_i, model_i
|
157 |
+
elif model_name == "RolmOCR": processor, model = processor_j, model_j
|
158 |
else:
|
159 |
yield "Invalid model selected.", "Invalid model selected.", None
|
160 |
return
|
|
|
224 |
"olmOCR-7B-0725",
|
225 |
"Nanonets-OCR-s",
|
226 |
"Megalodon-OCR-Sync-0713",
|
227 |
+
"RolmOCR"
|
228 |
],
|
229 |
label="Select Model", value="Nanonets-OCR-s"
|
230 |
)
|