Update app.py
Browse files
app.py
CHANGED
|
@@ -10,8 +10,6 @@ MODEL_PATH = "Qwen/Qwen3-VL-2B-Instruct"
|
|
| 10 |
CPU_DEVICE = "cpu"
|
| 11 |
|
| 12 |
# --- Model and Processor Loading ---
|
| 13 |
-
# This will be done once when the Space starts.
|
| 14 |
-
# 'device_map="auto"' will correctly assign the model to the CPU in this environment.
|
| 15 |
print("Loading model and processor... This will take a few minutes on a CPU.")
|
| 16 |
processor = AutoProcessor.from_pretrained(MODEL_PATH, trust_remote_code=True)
|
| 17 |
model = Qwen3VLForConditionalGeneration.from_pretrained(
|
|
@@ -81,7 +79,7 @@ with gr.Blocks() as demo:
|
|
| 81 |
"""
|
| 82 |
# Qwen3-VL-2B-Instruct CPU Demo
|
| 83 |
This Space runs the `Qwen/Qwen3-VL-2B-Instruct` model using the standard `transformers` library.
|
| 84 |
-
**Warning:** Running this on a free CPU Space is **very slow**.
|
| 85 |
"""
|
| 86 |
)
|
| 87 |
|
|
|
|
| 10 |
CPU_DEVICE = "cpu"
|
| 11 |
|
| 12 |
# --- Model and Processor Loading ---
|
|
|
|
|
|
|
| 13 |
print("Loading model and processor... This will take a few minutes on a CPU.")
|
| 14 |
processor = AutoProcessor.from_pretrained(MODEL_PATH, trust_remote_code=True)
|
| 15 |
model = Qwen3VLForConditionalGeneration.from_pretrained(
|
|
|
|
| 79 |
"""
|
| 80 |
# Qwen3-VL-2B-Instruct CPU Demo
|
| 81 |
This Space runs the `Qwen/Qwen3-VL-2B-Instruct` model using the standard `transformers` library.
|
| 82 |
+
**Warning:** Running this on a free CPU Space is **very slow**. Duplicate this space for solo experience.
|
| 83 |
"""
|
| 84 |
)
|
| 85 |
|