Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -48,7 +48,7 @@ controlnet = ControlNetModel.from_pretrained(
|
|
48 |
"lllyasviel/sd-controlnet-openpose", torch_dtype=torch.get_default_dtype()
|
49 |
)
|
50 |
pipe = StableDiffusionControlNetPipeline.from_pretrained(
|
51 |
-
"runwayml/stable-diffusion-v1-5", controlnet=controlnet, safety_checker=
|
52 |
)
|
53 |
pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
|
54 |
|
@@ -134,13 +134,4 @@ with gr.Blocks() as blocks:
|
|
134 |
_js=get_js_image)
|
135 |
blocks.load(None, None, None, _js=load_js)
|
136 |
|
137 |
-
gr.Examples(fn=generate_images,
|
138 |
-
examples=[
|
139 |
-
["./yoga1.jpeg",
|
140 |
-
"best quality, extremely detailed"]
|
141 |
-
],
|
142 |
-
inputs=[image_in_img, prompt],
|
143 |
-
outputs=[gallery],
|
144 |
-
cache_examples=True)
|
145 |
-
|
146 |
blocks.launch(debug=True)
|
|
|
48 |
"lllyasviel/sd-controlnet-openpose", torch_dtype=torch.get_default_dtype()
|
49 |
)
|
50 |
pipe = StableDiffusionControlNetPipeline.from_pretrained(
|
51 |
+
"runwayml/stable-diffusion-v1-5", controlnet=controlnet, safety_checker=True, torch_dtype=torch.get_default_dtype()
|
52 |
)
|
53 |
pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
|
54 |
|
|
|
134 |
_js=get_js_image)
|
135 |
blocks.load(None, None, None, _js=load_js)
|
136 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
137 |
blocks.launch(debug=True)
|