qq-hzlh commited on
Commit
b434b95
·
1 Parent(s): 03418d9

change model

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -116,7 +116,7 @@ def process_image_and_text(image, text):
116
  if __name__ == "__main__":
117
  import gradio as gr
118
 
119
- model_path = "omlab/VLM-R1-Qwen2.5VL-3B-Math-0305"
120
  # device = "cuda" if torch.cuda.is_available() else "cpu"
121
  device = "cuda"
122
  model = Qwen2_5_VLForConditionalGeneration.from_pretrained(model_path, torch_dtype=torch.float16)
 
116
  if __name__ == "__main__":
117
  import gradio as gr
118
 
119
+ model_path = "SZhanZ/Qwen2.5VL-VLM-R1-REC-step500"
120
  # device = "cuda" if torch.cuda.is_available() else "cpu"
121
  device = "cuda"
122
  model = Qwen2_5_VLForConditionalGeneration.from_pretrained(model_path, torch_dtype=torch.float16)