Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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
|
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
|
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
|
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 |
)
|