Spaces:
Running
on
Zero
Running
on
Zero
wew
Browse files
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",
|