Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -174,7 +174,8 @@ def _transform_messages(original_messages):
|
|
174 |
return transformed_messages
|
175 |
|
176 |
# --------- Configuration & Model Loading ---------
|
177 |
-
MODEL_DIR = "
|
|
|
178 |
# Load processor, tokenizer, model for Qwen2.5-VL
|
179 |
model = Qwen2_5_VLForConditionalGeneration.from_pretrained(
|
180 |
MODEL_DIR,
|
|
|
174 |
return transformed_messages
|
175 |
|
176 |
# --------- Configuration & Model Loading ---------
|
177 |
+
MODEL_DIR = "yejunliang23/ShapeLLM-7B-omni"
|
178 |
+
"Qwen/Qwen2.5-VL-3B-Instruct"
|
179 |
# Load processor, tokenizer, model for Qwen2.5-VL
|
180 |
model = Qwen2_5_VLForConditionalGeneration.from_pretrained(
|
181 |
MODEL_DIR,
|