Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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"
|
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 |
)
|