prithivMLmods commited on
Commit
8e118cf
·
verified ·
1 Parent(s): 96d5f01

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
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 VLM-R1-Qwen2.5VL-3B-OVD-0321
36
- MODEL_ID_X = "omlab/VLM-R1-Qwen2.5VL-3B-OVD-0321"
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 == "VLM-R1-Qwen2.5VL":
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 == "VLM-R1-Qwen2.5VL":
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=["Camel-Doc-OCR-062825", "GLM-4.1V-9B-Thinking", "Megalodon-OCR-Sync-0713", "DeepEyes-7B-Thinking", "VLM-R1-Qwen2.5VL"],
291
  label="Select Model",
292
- value="Camel-Doc-OCR-062825"
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)")