prithivMLmods commited on
Commit
2f849e9
·
verified ·
1 Parent(s): efd0961

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -91,8 +91,8 @@ model_i = Qwen2_5_VLForConditionalGeneration.from_pretrained(
91
 
92
  #-----------------------------------------#
93
 
94
- # Load Qwen2.5-VL-3B-Instruct
95
- MODEL_ID_J = "Qwen/Qwen2.5-VL-3B-Instruct"
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 == "Qwen2.5-VL-3B-Instruct": processor, model = processor_j, model_j
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
- "Qwen2.5-VL-3B-Instruct"
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
  )