yejunliang23 commited on
Commit
a97bc04
·
unverified ·
1 Parent(s): cc846d1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -174,7 +174,8 @@ def _transform_messages(original_messages):
174
  return transformed_messages
175
 
176
  # --------- Configuration & Model Loading ---------
177
- MODEL_DIR = "Qwen/Qwen2.5-VL-3B-Instruct"
 
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,