stzhao commited on
Commit
05c29cb
Β·
verified Β·
1 Parent(s): 719202e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -6,7 +6,7 @@ import time
6
  import torch
7
  import spaces
8
 
9
- MODEL_ID = "Qwen/Qwen2.5-VL-7B-Instruct"
10
  processor = AutoProcessor.from_pretrained(MODEL_ID, trust_remote_code=True)
11
  model = Qwen2_5_VLForConditionalGeneration.from_pretrained(
12
  MODEL_ID,
@@ -86,7 +86,7 @@ examples = [
86
 
87
  demo = gr.ChatInterface(
88
  fn=model_inference,
89
- description="# **Qwen2.5-VL-7B-Instruct**",
90
  examples=examples,
91
  textbox=gr.MultimodalTextbox(label="Query Input", file_types=["image"], file_count="multiple"),
92
  stop_btn="Stop Generation",
 
6
  import torch
7
  import spaces
8
 
9
+ MODEL_ID = "Qwen/Qwen2.5-VL-3B-Instruct"
10
  processor = AutoProcessor.from_pretrained(MODEL_ID, trust_remote_code=True)
11
  model = Qwen2_5_VLForConditionalGeneration.from_pretrained(
12
  MODEL_ID,
 
86
 
87
  demo = gr.ChatInterface(
88
  fn=model_inference,
89
+ description="# **Qwen2.5-VL-3B-Instruct**",
90
  examples=examples,
91
  textbox=gr.MultimodalTextbox(label="Query Input", file_types=["image"], file_count="multiple"),
92
  stop_btn="Stop Generation",