Spaces:

Heartsync
/

NSFW-detection

Running on Zero

seawolf2357 commited on May 5

Commit

1256204

verified ·

1 Parent(s): ce843ba

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -19,7 +19,7 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 # STABLE DIFFUSION XL PIPELINE
 # ------------------------------------------------------------
 pipe = StableDiffusionXLPipeline.from_pretrained(
-    "votepurchase/waiNSFWIllustrious_v120",
     torch_dtype=torch.float16,
     variant="fp16",
     use_safetensors=True,

 # STABLE DIFFUSION XL PIPELINE
 # ------------------------------------------------------------
 pipe = StableDiffusionXLPipeline.from_pretrained(
+    "Heartsync/NSFW-Uncensored",
     torch_dtype=torch.float16,
     variant="fp16",
     use_safetensors=True,