Spaces:
Runtime error
Runtime error
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)
|