Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -66,8 +66,8 @@ model_g = Qwen2_5_VLForConditionalGeneration.from_pretrained(
|
|
66 |
).to(device).eval()
|
67 |
#-----------------------------subfolder-----------------------------#
|
68 |
|
69 |
-
# Load
|
70 |
-
MODEL_ID_O = "prithivMLmods/
|
71 |
processor_o = AutoProcessor.from_pretrained(MODEL_ID_O, trust_remote_code=True)
|
72 |
model_o = Qwen2_5_VLForConditionalGeneration.from_pretrained(
|
73 |
MODEL_ID_O,
|
@@ -116,7 +116,7 @@ def generate_image(model_name: str, text: str, image: Image.Image,
|
|
116 |
elif model_name == "MonkeyOCR-Recognition":
|
117 |
processor = processor_g
|
118 |
model = model_g
|
119 |
-
elif model_name == "
|
120 |
processor = processor_o
|
121 |
model = model_o
|
122 |
else:
|
@@ -174,7 +174,7 @@ def generate_video(model_name: str, text: str, video_path: str,
|
|
174 |
elif model_name == "MonkeyOCR-Recognition":
|
175 |
processor = processor_g
|
176 |
model = model_g
|
177 |
-
elif model_name == "
|
178 |
processor = processor_o
|
179 |
model = model_o
|
180 |
else:
|
@@ -290,7 +290,7 @@ with gr.Blocks(css=css, theme="bethecloud/storj_theme") as demo:
|
|
290 |
markdown_output = gr.Markdown(label="(Result.Md)")
|
291 |
|
292 |
model_choice = gr.Radio(
|
293 |
-
choices=["docscopeOCR-7B-050425-exp", "MonkeyOCR-Recognition", "coreOCR-7B-050325-preview"
|
294 |
label="Select Model",
|
295 |
value="docscopeOCR-7B-050425-exp"
|
296 |
)
|
|
|
66 |
).to(device).eval()
|
67 |
#-----------------------------subfolder-----------------------------#
|
68 |
|
69 |
+
# Load Camel-Doc-OCR-080125
|
70 |
+
MODEL_ID_O = "prithivMLmods/Camel-Doc-OCR-080125"
|
71 |
processor_o = AutoProcessor.from_pretrained(MODEL_ID_O, trust_remote_code=True)
|
72 |
model_o = Qwen2_5_VLForConditionalGeneration.from_pretrained(
|
73 |
MODEL_ID_O,
|
|
|
116 |
elif model_name == "MonkeyOCR-Recognition":
|
117 |
processor = processor_g
|
118 |
model = model_g
|
119 |
+
elif model_name == "Camel-Doc-OCR-080125(v2)":
|
120 |
processor = processor_o
|
121 |
model = model_o
|
122 |
else:
|
|
|
174 |
elif model_name == "MonkeyOCR-Recognition":
|
175 |
processor = processor_g
|
176 |
model = model_g
|
177 |
+
elif model_name == "Camel-Doc-OCR-080125(v2)":
|
178 |
processor = processor_o
|
179 |
model = model_o
|
180 |
else:
|
|
|
290 |
markdown_output = gr.Markdown(label="(Result.Md)")
|
291 |
|
292 |
model_choice = gr.Radio(
|
293 |
+
choices=["Camel-Doc-OCR-080125(v2)", "docscopeOCR-7B-050425-exp", "MonkeyOCR-Recognition", "coreOCR-7B-050325-preview"],
|
294 |
label="Select Model",
|
295 |
value="docscopeOCR-7B-050425-exp"
|
296 |
)
|