Spaces:
Running on Zero
Running on Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -32,8 +32,8 @@ model_m = Qwen2_5_VLForConditionalGeneration.from_pretrained(
|
|
| 32 |
torch_dtype=torch.float16
|
| 33 |
).to(device).eval()
|
| 34 |
|
| 35 |
-
# Load
|
| 36 |
-
MODEL_ID_X = "
|
| 37 |
processor_x = AutoProcessor.from_pretrained(MODEL_ID_X, trust_remote_code=True)
|
| 38 |
model_x = Qwen2_5_VLForConditionalGeneration.from_pretrained(
|
| 39 |
MODEL_ID_X,
|
|
@@ -110,7 +110,7 @@ def generate_image(model_name: str, text: str, image: Image.Image,
|
|
| 110 |
elif model_name == "DeepEyes-7B-Thinking":
|
| 111 |
processor = processor_y
|
| 112 |
model = model_y
|
| 113 |
-
elif model_name == "
|
| 114 |
processor = processor_x
|
| 115 |
model = model_x
|
| 116 |
else:
|
|
@@ -169,7 +169,7 @@ def generate_video(model_name: str, text: str, video_path: str,
|
|
| 169 |
elif model_name == "DeepEyes-7B-Thinking":
|
| 170 |
processor = processor_y
|
| 171 |
model = model_y
|
| 172 |
-
elif model_name == "
|
| 173 |
processor = processor_x
|
| 174 |
model = model_x
|
| 175 |
else:
|
|
@@ -287,9 +287,9 @@ with gr.Blocks(css=css, theme="bethecloud/storj_theme") as demo:
|
|
| 287 |
markdown_output = gr.Markdown(label="(Result.md)")
|
| 288 |
|
| 289 |
model_choice = gr.Radio(
|
| 290 |
-
choices=["
|
| 291 |
label="Select Model",
|
| 292 |
-
value="
|
| 293 |
)
|
| 294 |
|
| 295 |
gr.Markdown("**Model Info 💻** | [Report Bug](https://huggingface.co/spaces/prithivMLmods/Multimodal-OCR-Comparator/discussions)")
|
|
|
|
| 32 |
torch_dtype=torch.float16
|
| 33 |
).to(device).eval()
|
| 34 |
|
| 35 |
+
# Load Qwen2.5-VL-3B-Instruct-abliterated
|
| 36 |
+
MODEL_ID_X = "huihui-ai/Qwen2.5-VL-3B-Instruct-abliterated"
|
| 37 |
processor_x = AutoProcessor.from_pretrained(MODEL_ID_X, trust_remote_code=True)
|
| 38 |
model_x = Qwen2_5_VLForConditionalGeneration.from_pretrained(
|
| 39 |
MODEL_ID_X,
|
|
|
|
| 110 |
elif model_name == "DeepEyes-7B-Thinking":
|
| 111 |
processor = processor_y
|
| 112 |
model = model_y
|
| 113 |
+
elif model_name == "Qwen2.5-VL-3B-Instruct-abliterated":
|
| 114 |
processor = processor_x
|
| 115 |
model = model_x
|
| 116 |
else:
|
|
|
|
| 169 |
elif model_name == "DeepEyes-7B-Thinking":
|
| 170 |
processor = processor_y
|
| 171 |
model = model_y
|
| 172 |
+
elif model_name == "Qwen2.5-VL-3B-Instruct-abliterated":
|
| 173 |
processor = processor_x
|
| 174 |
model = model_x
|
| 175 |
else:
|
|
|
|
| 287 |
markdown_output = gr.Markdown(label="(Result.md)")
|
| 288 |
|
| 289 |
model_choice = gr.Radio(
|
| 290 |
+
choices=["GLM-4.1V-9B-Thinking", "Camel-Doc-OCR-062825", "Megalodon-OCR-Sync-0713", "DeepEyes-7B-Thinking", "Qwen2.5-VL-3B-Instruct-abliterated"],
|
| 291 |
label="Select Model",
|
| 292 |
+
value="GLM-4.1V-9B-Thinking"
|
| 293 |
)
|
| 294 |
|
| 295 |
gr.Markdown("**Model Info 💻** | [Report Bug](https://huggingface.co/spaces/prithivMLmods/Multimodal-OCR-Comparator/discussions)")
|