Hugging Face's logo

Spaces:
geshang
/
Seg-R1-demo
Runtime error

App Files Community
1
geshang commited on
Commit
7d7e76b
·
verified ·
1 Parent(s): 1b8a58c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -61,7 +61,7 @@ from sam2.sam2_image_predictor import SAM2ImagePredictor
61
  print("🎉 SAM2 modules imported successfully!")
62
 
63
 
64
- MODEL_PATH = "geshang/Seg-R1-3B"
65
  SAM_CHECKPOINT = "sam2_weights/sam2.1_hiera_large.pt"
66
 
67
  DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 
61
  print("🎉 SAM2 modules imported successfully!")
62
 
63
 
64
+ MODEL_PATH = "geshang/Seg-R1-7B"
65
  SAM_CHECKPOINT = "sam2_weights/sam2.1_hiera_large.pt"
66
 
67
  DEVICE = "cuda" if torch.cuda.is_available() else "cpu"