Spaces:
Running
on
Zero
Running
on
Zero
change model
Browse files
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 = "
|
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)
|