John Ho commited on
Commit
110f151
·
1 Parent(s): 6dd8fb2

trying newer torch versions

Browse files
Files changed (2) hide show
  1. app_qwen25vl.py +5 -1
  2. pyproject.toml +2 -1
app_qwen25vl.py CHANGED
@@ -249,6 +249,10 @@ with gr.Blocks(css=css) as demo:
249
  label="Text Prompt",
250
  value="Describe the camera motion in this video.",
251
  )
 
 
 
 
252
  submit_btn = gr.Button(value="Submit")
253
  with gr.Column():
254
  output_text = gr.Textbox(label="Output Text", interactive=False)
@@ -258,7 +262,7 @@ with gr.Blocks(css=css) as demo:
258
 
259
  submit_btn.click(
260
  run_example,
261
- [input_media, text_input],
262
  [output_text, time_taken],
263
  ) # Ensure output components match yield order
264
 
 
249
  label="Text Prompt",
250
  value="Describe the camera motion in this video.",
251
  )
252
+ model_selector = gr.Dropdown(
253
+ choices=list(models.keys()),
254
+ label="Model",
255
+ )
256
  submit_btn = gr.Button(value="Submit")
257
  with gr.Column():
258
  output_text = gr.Textbox(label="Output Text", interactive=False)
 
262
 
263
  submit_btn.click(
264
  run_example,
265
+ [input_media, text_input, model_selector],
266
  [output_text, time_taken],
267
  ) # Ensure output components match yield order
268
 
pyproject.toml CHANGED
@@ -10,7 +10,8 @@ dependencies = [
10
  "pydantic==2.10.6",
11
  "loguru>=0.7.3",
12
  "qwen-vl-utils>=0.0.11",
13
- "torchvision==0.19.0",
 
14
  "ffmpeg-python>=0.2.0",
15
  "accelerate==0.32.1",
16
  "bitsandbytes>=0.41.3",
 
10
  "pydantic==2.10.6",
11
  "loguru>=0.7.3",
12
  "qwen-vl-utils>=0.0.11",
13
+ "torchvision>=0.22.0",
14
+ "torch>=2.7.0",
15
  "ffmpeg-python>=0.2.0",
16
  "accelerate==0.32.1",
17
  "bitsandbytes>=0.41.3",