Hugging Face's logo

Spaces:
geshang
/
Seg-R1-demo
Runtime error

App Files Community
1
geshang commited on
Commit
188e26a
·
verified ·
1 Parent(s): b23a905

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -82,8 +82,8 @@ except Exception as e:
82
  processor = None
83
 
84
  # SAM Wrapper
 
85
  class CustomSAMWrapper:
86
- @spaces.GPU
87
  def __init__(self, model_path: str, device: str = DEVICE):
88
  # try:
89
  self.device = torch.device(device)
 
82
  processor = None
83
 
84
  # SAM Wrapper
85
+ @spaces.GPU
86
  class CustomSAMWrapper:
 
87
  def __init__(self, model_path: str, device: str = DEVICE):
88
  # try:
89
  self.device = torch.device(device)