Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
|
@@ -29,6 +29,7 @@ def predict(dict, prompt="", negative_prompt="", guidance_scale=7.5, steps=30, s
|
|
| 29 |
pipe = AutoPipelineForInpainting.from_pretrained("emilianJR/epiCRealism", vae=vae).to("cuda")
|
| 30 |
if model == "Realistic_V6.0":
|
| 31 |
pipe = AutoPipelineForInpainting.from_pretrained("SG161222/Realistic_Vision_V6.0_B1_noVAE", vae=vae).to("cuda")
|
|
|
|
| 32 |
|
| 33 |
|
| 34 |
if negative_prompt == "":
|
|
|
|
| 29 |
pipe = AutoPipelineForInpainting.from_pretrained("emilianJR/epiCRealism", vae=vae).to("cuda")
|
| 30 |
if model == "Realistic_V6.0":
|
| 31 |
pipe = AutoPipelineForInpainting.from_pretrained("SG161222/Realistic_Vision_V6.0_B1_noVAE", vae=vae).to("cuda")
|
| 32 |
+
pipe.safety_checker = lambda images, **kwargs: (images, [False] * len(images))
|
| 33 |
|
| 34 |
|
| 35 |
if negative_prompt == "":
|