stzhao commited on
Commit
1cff9f0
Β·
verified Β·
1 Parent(s): 384c558

Update app.py

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