Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -59,7 +59,7 @@ SYSTEM_PROMPT = (
|
|
| 59 |
|
| 60 |
|
| 61 |
# Load model and processor
|
| 62 |
-
processor = AutoProcessor.from_pretrained("JosephZ/
|
| 63 |
|
| 64 |
device='cuda' if torch.cuda.is_available() else "cpu"
|
| 65 |
model = Qwen2VLForConditionalGeneration.from_pretrained("JosephZ/qwen2vl-7b-sft-grpo-close-sgg",
|
|
|
|
| 59 |
|
| 60 |
|
| 61 |
# Load model and processor
|
| 62 |
+
processor = AutoProcessor.from_pretrained("JosephZ/R1-SGG-7B", max_pixels=1024*28*28)
|
| 63 |
|
| 64 |
device='cuda' if torch.cuda.is_available() else "cpu"
|
| 65 |
model = Qwen2VLForConditionalGeneration.from_pretrained("JosephZ/qwen2vl-7b-sft-grpo-close-sgg",
|