prithivMLmods commited on
Commit
1a13030
·
verified ·
1 Parent(s): 15b057f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -16
app.py CHANGED
@@ -61,15 +61,6 @@ model_s = Glm4vForConditionalGeneration.from_pretrained(
61
  torch_dtype=torch.float16
62
  ).to(device).eval()
63
 
64
- # Load OVR-7B-RL
65
- MODEL_ID_F = "Kangheng/OVR-7B-RL"
66
- processor_f = AutoProcessor.from_pretrained(MODEL_ID_F, trust_remote_code=True)
67
- model_f = Qwen2_5_VLForConditionalGeneration.from_pretrained(
68
- MODEL_ID_F,
69
- trust_remote_code=True,
70
- torch_dtype=torch.float16
71
- ).to(device).eval()
72
-
73
  # Load DeepEyes-7B
74
  MODEL_ID_Y = "ChenShawn/DeepEyes-7B"
75
  processor_y = AutoProcessor.from_pretrained(MODEL_ID_Y, trust_remote_code=True)
@@ -121,9 +112,6 @@ def generate_image(model_name: str, text: str, image: Image.Image,
121
  elif model_name == "GLM-4.1V-9B-Thinking":
122
  processor = processor_s
123
  model = model_s
124
- elif model_name == "OVR-7B-RL":
125
- processor = processor_f
126
- model = model_f
127
  elif model_name == "DeepEyes-7B-Thinking":
128
  processor = processor_y
129
  model = model_y
@@ -183,9 +171,6 @@ def generate_video(model_name: str, text: str, video_path: str,
183
  elif model_name == "GLM-4.1V-9B-Thinking":
184
  processor = processor_s
185
  model = model_s
186
- elif model_name == "OVR-7B-RL":
187
- processor = processor_f
188
- model = model_f
189
  elif model_name == "DeepEyes-7B-Thinking":
190
  processor = processor_y
191
  model = model_y
@@ -304,7 +289,7 @@ with gr.Blocks(css=css, theme="bethecloud/storj_theme") as demo:
304
  markdown_output = gr.Markdown(label="(Result.md)")
305
 
306
  model_choice = gr.Radio(
307
- choices=["Camel-Doc-OCR-062825", "GLM-4.1V-9B-Thinking", "Megalodon-OCR-Sync-0713", "MonkeyOCR-pro-1.2B", "DeepEyes-7B-Thinking", "OVR-7B-RL"],
308
  label="Select Model",
309
  value="Camel-Doc-OCR-062825"
310
  )
 
61
  torch_dtype=torch.float16
62
  ).to(device).eval()
63
 
 
 
 
 
 
 
 
 
 
64
  # Load DeepEyes-7B
65
  MODEL_ID_Y = "ChenShawn/DeepEyes-7B"
66
  processor_y = AutoProcessor.from_pretrained(MODEL_ID_Y, trust_remote_code=True)
 
112
  elif model_name == "GLM-4.1V-9B-Thinking":
113
  processor = processor_s
114
  model = model_s
 
 
 
115
  elif model_name == "DeepEyes-7B-Thinking":
116
  processor = processor_y
117
  model = model_y
 
171
  elif model_name == "GLM-4.1V-9B-Thinking":
172
  processor = processor_s
173
  model = model_s
 
 
 
174
  elif model_name == "DeepEyes-7B-Thinking":
175
  processor = processor_y
176
  model = model_y
 
289
  markdown_output = gr.Markdown(label="(Result.md)")
290
 
291
  model_choice = gr.Radio(
292
+ choices=["Camel-Doc-OCR-062825", "GLM-4.1V-9B-Thinking", "Megalodon-OCR-Sync-0713", "MonkeyOCR-pro-1.2B", "DeepEyes-7B-Thinking"],
293
  label="Select Model",
294
  value="Camel-Doc-OCR-062825"
295
  )