amaye15 commited on
Commit
af0381c
1 Parent(s): c8a6ab0

Local & Remote fully working now

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -18,7 +18,10 @@ PREDICTOR = SAM2ImagePredictor.from_pretrained(MODEL, device=DEVICE)
18
 
19
  DESTINATION_DS = "amaye15/object-segmentation"
20
 
21
- login(os.getenv("TOKEN"))
 
 
 
22
 
23
  IMAGE = None
24
  MASKS = None
 
18
 
19
  DESTINATION_DS = "amaye15/object-segmentation"
20
 
21
+
22
+ token = os.getenv("TOKEN")
23
+ if token:
24
+ login(token)
25
 
26
  IMAGE = None
27
  MASKS = None