Spaces:
Running
on
Zero
Running
on
Zero
update app
Browse files
app.py
CHANGED
|
@@ -118,8 +118,8 @@ model_x = Qwen2_5_VLForConditionalGeneration.from_pretrained(
|
|
| 118 |
torch_dtype=torch.float16
|
| 119 |
).to(device).eval()
|
| 120 |
|
| 121 |
-
# Load Qwen3-VL-
|
| 122 |
-
MODEL_ID_Q = "Qwen/Qwen3-VL-
|
| 123 |
processor_q = AutoProcessor.from_pretrained(MODEL_ID_Q, trust_remote_code=True)
|
| 124 |
model_q = Qwen3VLForConditionalGeneration.from_pretrained(
|
| 125 |
MODEL_ID_Q,
|
|
@@ -163,7 +163,7 @@ def generate_image(model_name: str, text: str, image: Image.Image,
|
|
| 163 |
processor, model = processor_m, model_m
|
| 164 |
elif model_name == "Qwen2.5-VL-3B-Instruct":
|
| 165 |
processor, model = processor_x, model_x
|
| 166 |
-
elif model_name == "Qwen3-VL-
|
| 167 |
processor, model = processor_q, model_q
|
| 168 |
else:
|
| 169 |
yield "Invalid model selected.", "Invalid model selected."
|
|
@@ -201,7 +201,7 @@ def generate_video(model_name: str, text: str, video_path: str,
|
|
| 201 |
processor, model = processor_m, model_m
|
| 202 |
elif model_name == "Qwen2.5-VL-3B-Instruct":
|
| 203 |
processor, model = processor_x, model_x
|
| 204 |
-
elif model_name == "Qwen3-VL-
|
| 205 |
processor, model = processor_q, model_q
|
| 206 |
else:
|
| 207 |
yield "Invalid model selected.", "Invalid model selected."
|
|
@@ -297,9 +297,9 @@ with gr.Blocks(css=css, theme=steel_blue_theme) as demo:
|
|
| 297 |
markdown_output = gr.Markdown()
|
| 298 |
|
| 299 |
model_choice = gr.Radio(
|
| 300 |
-
choices=["Qwen3-VL-
|
| 301 |
label="Select Model",
|
| 302 |
-
value="Qwen3-VL-
|
| 303 |
)
|
| 304 |
|
| 305 |
image_submit.click(
|
|
|
|
| 118 |
torch_dtype=torch.float16
|
| 119 |
).to(device).eval()
|
| 120 |
|
| 121 |
+
# Load Qwen3-VL-2B-Instruct
|
| 122 |
+
MODEL_ID_Q = "Qwen/Qwen3-VL-2B-Instruct"
|
| 123 |
processor_q = AutoProcessor.from_pretrained(MODEL_ID_Q, trust_remote_code=True)
|
| 124 |
model_q = Qwen3VLForConditionalGeneration.from_pretrained(
|
| 125 |
MODEL_ID_Q,
|
|
|
|
| 163 |
processor, model = processor_m, model_m
|
| 164 |
elif model_name == "Qwen2.5-VL-3B-Instruct":
|
| 165 |
processor, model = processor_x, model_x
|
| 166 |
+
elif model_name == "Qwen3-VL-2B-Instruct":
|
| 167 |
processor, model = processor_q, model_q
|
| 168 |
else:
|
| 169 |
yield "Invalid model selected.", "Invalid model selected."
|
|
|
|
| 201 |
processor, model = processor_m, model_m
|
| 202 |
elif model_name == "Qwen2.5-VL-3B-Instruct":
|
| 203 |
processor, model = processor_x, model_x
|
| 204 |
+
elif model_name == "Qwen3-VL-2B-Instruct":
|
| 205 |
processor, model = processor_q, model_q
|
| 206 |
else:
|
| 207 |
yield "Invalid model selected.", "Invalid model selected."
|
|
|
|
| 297 |
markdown_output = gr.Markdown()
|
| 298 |
|
| 299 |
model_choice = gr.Radio(
|
| 300 |
+
choices=["Qwen3-VL-2B-Instruct", "Qwen2.5-VL-3B-Instruct", "Qwen2.5-VL-7B-Instruct"],
|
| 301 |
label="Select Model",
|
| 302 |
+
value="Qwen3-VL-2B-Instruct"
|
| 303 |
)
|
| 304 |
|
| 305 |
image_submit.click(
|