Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -200,15 +200,14 @@ class StableDiffusionInpaintingPipeline(DiffusionPipeline):
|
|
200 |
|
201 |
return {"sample": image, "nsfw_content_detected": has_nsfw_concept}
|
202 |
|
203 |
-
|
204 |
-
|
205 |
|
206 |
-
|
207 |
model_path,
|
208 |
revision="fp16",
|
209 |
torch_dtype=torch.float16,
|
210 |
-
use_auth_token=True
|
211 |
-
).to(device)
|
212 |
|
213 |
import gdown
|
214 |
def download_gdrive_url():
|
|
|
200 |
|
201 |
return {"sample": image, "nsfw_content_detected": has_nsfw_concept}
|
202 |
|
203 |
+
device = "cuda"
|
204 |
+
model_path = "CompVis/stable-diffusion-v1-4"
|
205 |
|
206 |
+
pipe = StableDiffusionInpaintingPipeline.from_pretrained(
|
207 |
model_path,
|
208 |
revision="fp16",
|
209 |
torch_dtype=torch.float16,
|
210 |
+
use_auth_token=True).to(device)
|
|
|
211 |
|
212 |
import gdown
|
213 |
def download_gdrive_url():
|