prithivMLmods commited on
Commit
8c40983
·
verified ·
1 Parent(s): 01b6b5c

update app

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -88,12 +88,12 @@ DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
88
  DTYPE = "auto"
89
 
90
  qwen_model = Qwen3VLForConditionalGeneration.from_pretrained(
91
- "Qwen/Qwen3-VL-32B-Instruct",
92
  dtype=DTYPE,
93
  device_map=DEVICE,
94
  ).eval()
95
  qwen_processor = Qwen3VLProcessor.from_pretrained(
96
- "Qwen/Qwen3-VL-32B-Instruct",
97
  )
98
 
99
  CATEGORIES = ["Query", "Caption", "Point", "Detect"]
 
88
  DTYPE = "auto"
89
 
90
  qwen_model = Qwen3VLForConditionalGeneration.from_pretrained(
91
+ "Qwen/Qwen3-VL-8B-Instruct",
92
  dtype=DTYPE,
93
  device_map=DEVICE,
94
  ).eval()
95
  qwen_processor = Qwen3VLProcessor.from_pretrained(
96
+ "Qwen/Qwen3-VL-8B-Instruct",
97
  )
98
 
99
  CATEGORIES = ["Query", "Caption", "Point", "Detect"]