LPX55 commited on
Commit
1d11eb2
·
1 Parent(s): 6bc4886
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ from sam2.sam2_image_predictor import SAM2ImagePredictor
14
 
15
  DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
16
  SAM_MODEL = "facebook/sam2.1-hiera-large"
17
-
18
  MODELS = {
19
  "RealVisXL V5.0 Lightning": "SG161222/RealVisXL_V5.0_Lightning",
20
  "Lustify Lightning": "GraydientPlatformAPI/lustify-lightning",
 
14
 
15
  DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
16
  SAM_MODEL = "facebook/sam2.1-hiera-large"
17
+ PREDICTOR = None
18
  MODELS = {
19
  "RealVisXL V5.0 Lightning": "SG161222/RealVisXL_V5.0_Lightning",
20
  "Lustify Lightning": "GraydientPlatformAPI/lustify-lightning",