Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -40,12 +40,12 @@ def resize_image(input_path, output_path, target_height):
|
|
40 |
|
41 |
def load_pipeline(control_type):
|
42 |
if control_type == "canny":
|
43 |
-
|
44 |
"stabilityai/stable-diffusion-3-medium-diffusers",
|
45 |
controlnet=controlnet_canny
|
46 |
)
|
47 |
elif control_type == "tile":
|
48 |
-
|
49 |
"stabilityai/stable-diffusion-3-medium-diffusers",
|
50 |
controlnet=controlnet_tile
|
51 |
)
|
|
|
40 |
|
41 |
def load_pipeline(control_type):
|
42 |
if control_type == "canny":
|
43 |
+
pipe_canny = StableDiffusion3ControlNetPipeline.from_pretrained(
|
44 |
"stabilityai/stable-diffusion-3-medium-diffusers",
|
45 |
controlnet=controlnet_canny
|
46 |
)
|
47 |
elif control_type == "tile":
|
48 |
+
pipe_tile = StableDiffusion3ControlNetPipeline.from_pretrained(
|
49 |
"stabilityai/stable-diffusion-3-medium-diffusers",
|
50 |
controlnet=controlnet_tile
|
51 |
)
|