prithivMLmods commited on
Commit
aa59ead
·
verified ·
1 Parent(s): 5fa0137

update app

Browse files
Files changed (1) hide show
  1. app.py +6 -6
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-4B-Instruct
122
- MODEL_ID_Q = "Qwen/Qwen3-VL-4B-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,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-4B-Instruct":
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-4B-Instruct":
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-4B-Instruct", "Qwen2.5-VL-3B-Instruct", "Qwen2.5-VL-7B-Instruct"],
301
  label="Select Model",
302
- value="Qwen3-VL-4B-Instruct"
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(