Spaces:
Running
on
Zero
Running
on
Zero
update app
Browse files
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-
|
| 92 |
dtype=DTYPE,
|
| 93 |
device_map=DEVICE,
|
| 94 |
).eval()
|
| 95 |
qwen_processor = Qwen3VLProcessor.from_pretrained(
|
| 96 |
-
"Qwen/Qwen3-VL-
|
| 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"]
|