prithivMLmods commited on
Commit
5527e7f
·
verified ·
1 Parent(s): 1717174

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -100,7 +100,7 @@ def generate_image(model_name: str, text: str, image: Image.Image,
100
  if model_name == "RolmOCR":
101
  processor = processor_m
102
  model = model_m
103
- elif model_name == "Qwen2-VL-OCR-2B-Instruct":
104
  processor = processor_x
105
  model = model_x
106
  elif model_name == "Nanonets-OCR-s":
@@ -158,7 +158,7 @@ def generate_video(model_name: str, text: str, video_path: str,
158
  if model_name == "RolmOCR":
159
  processor = processor_m
160
  model = model_m
161
- elif model_name == "Qwen2-VL-OCR-2B-Instruct":
162
  processor = processor_x
163
  model = model_x
164
  elif model_name == "Nanonets-OCR-s":
@@ -276,7 +276,7 @@ with gr.Blocks(css=css, theme="bethecloud/storj_theme") as demo:
276
  with gr.Accordion("Formatted Result (Result.md)", open=False):
277
  markdown_output = gr.Markdown(label="Formatted Result (Result.Md)")
278
  model_choice = gr.Radio(
279
- choices=["Nanonets-OCR-s", "Qwen2-VL-OCR-2B-Instruct", "RolmOCR", "Aya-Vision"],
280
  label="Select Model",
281
  value="Nanonets-OCR-s"
282
  )
 
100
  if model_name == "RolmOCR":
101
  processor = processor_m
102
  model = model_m
103
+ elif model_name == "Qwen2-VL-OCR":
104
  processor = processor_x
105
  model = model_x
106
  elif model_name == "Nanonets-OCR-s":
 
158
  if model_name == "RolmOCR":
159
  processor = processor_m
160
  model = model_m
161
+ elif model_name == "Qwen2-VL-OCR":
162
  processor = processor_x
163
  model = model_x
164
  elif model_name == "Nanonets-OCR-s":
 
276
  with gr.Accordion("Formatted Result (Result.md)", open=False):
277
  markdown_output = gr.Markdown(label="Formatted Result (Result.Md)")
278
  model_choice = gr.Radio(
279
+ choices=["Nanonets-OCR-s", "Qwen2-VL-OCR", "RolmOCR", "Aya-Vision"],
280
  label="Select Model",
281
  value="Nanonets-OCR-s"
282
  )