Update app_demo.py
Browse files- app_demo.py +1 -1
app_demo.py
CHANGED
|
@@ -47,7 +47,7 @@ pipe.to("cpu")
|
|
| 47 |
|
| 48 |
|
| 49 |
|
| 50 |
-
pipe = StableDiffusionPipeline.from_pretrained(model_id)
|
| 51 |
prompt = "A futuristic cityscape at sunset"
|
| 52 |
image = pipe(prompt).images[0]
|
| 53 |
image.show()
|
|
|
|
| 47 |
|
| 48 |
|
| 49 |
|
| 50 |
+
pipe = StableDiffusionPipeline.from_pretrained(model_id, safety_checker= None)
|
| 51 |
prompt = "A futuristic cityscape at sunset"
|
| 52 |
image = pipe(prompt).images[0]
|
| 53 |
image.show()
|