Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -103,7 +103,7 @@ def generate_image(model_name: str, text: str, image: Image.Image,
|
|
| 103 |
elif model_name == "coreOCR-7B-050325-preview":
|
| 104 |
processor = processor_x
|
| 105 |
model = model_x
|
| 106 |
-
elif model_name == "MonkeyOCR":
|
| 107 |
processor = processor_g
|
| 108 |
model = model_g
|
| 109 |
else:
|
|
@@ -157,7 +157,7 @@ def generate_video(model_name: str, text: str, video_path: str,
|
|
| 157 |
elif model_name == "coreOCR-7B-050325-preview":
|
| 158 |
processor = processor_x
|
| 159 |
model = model_x
|
| 160 |
-
elif model_name == "MonkeyOCR":
|
| 161 |
processor = processor_g
|
| 162 |
model = model_g
|
| 163 |
else:
|
|
@@ -258,7 +258,7 @@ with gr.Blocks(css=css, theme="bethecloud/storj_theme") as demo:
|
|
| 258 |
with gr.Column():
|
| 259 |
output = gr.Textbox(label="Output", interactive=False, lines=2, scale=2)
|
| 260 |
model_choice = gr.Radio(
|
| 261 |
-
choices=["docscopeOCR-7B-050425-exp", "MonkeyOCR", "coreOCR-7B-050325-preview"],
|
| 262 |
label="Select Model",
|
| 263 |
value="docscopeOCR-7B-050425-exp"
|
| 264 |
)
|
|
|
|
| 103 |
elif model_name == "coreOCR-7B-050325-preview":
|
| 104 |
processor = processor_x
|
| 105 |
model = model_x
|
| 106 |
+
elif model_name == "MonkeyOCR-Recognition":
|
| 107 |
processor = processor_g
|
| 108 |
model = model_g
|
| 109 |
else:
|
|
|
|
| 157 |
elif model_name == "coreOCR-7B-050325-preview":
|
| 158 |
processor = processor_x
|
| 159 |
model = model_x
|
| 160 |
+
elif model_name == "MonkeyOCR-Recognition":
|
| 161 |
processor = processor_g
|
| 162 |
model = model_g
|
| 163 |
else:
|
|
|
|
| 258 |
with gr.Column():
|
| 259 |
output = gr.Textbox(label="Output", interactive=False, lines=2, scale=2)
|
| 260 |
model_choice = gr.Radio(
|
| 261 |
+
choices=["docscopeOCR-7B-050425-exp", "MonkeyOCR-Recognition", "coreOCR-7B-050325-preview"],
|
| 262 |
label="Select Model",
|
| 263 |
value="docscopeOCR-7B-050425-exp"
|
| 264 |
)
|