Hugging Face's logo

Spaces:
geshang
/
Seg-R1-demo
Runtime error

App Files Community
1
geshang commited on
Commit
500db39
·
verified ·
1 Parent(s): 691579e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -65,7 +65,7 @@ print("🎉 SAM2 modules imported successfully!")
65
  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:
 
65
  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: