Hugging Face's logo

Spaces:
geshang
/
Seg-R1-demo
Running on Zero

App Files Community
geshang commited on
Commit
52be1b4
·
verified ·
1 Parent(s): 79be99e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -66,7 +66,7 @@ MODEL_PATH = "geshang/Seg-R1-7B"
66
  SAM_CHECKPOINT = "sam2_weights/sam2.1_hiera_large.pt"
67
 
68
  DEVICE = "cuda" #if torch.cuda.is_available() else "cpu"
69
- RESIZE_SIZE = (768, 768)
70
 
71
  try:
72
  model = Qwen2_5_VLForConditionalGeneration.from_pretrained(
 
66
  SAM_CHECKPOINT = "sam2_weights/sam2.1_hiera_large.pt"
67
 
68
  DEVICE = "cuda" #if torch.cuda.is_available() else "cpu"
69
+ RESIZE_SIZE = (1024, 1024)
70
 
71
  try:
72
  model = Qwen2_5_VLForConditionalGeneration.from_pretrained(