Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -58,7 +58,7 @@ def infer(image_in, prompt, inference_steps, guidance_scale, control_weight):
|
|
58 |
|
59 |
n_prompt = 'NSFW, nude, naked, porn, ugly'
|
60 |
|
61 |
-
|
62 |
|
63 |
# Canny preprocessing
|
64 |
image_to_canny = load_image(image_in)
|
@@ -89,6 +89,10 @@ def infer(image_in, prompt, inference_steps, guidance_scale, control_weight):
|
|
89 |
height=h
|
90 |
).images[0]
|
91 |
|
|
|
|
|
|
|
|
|
92 |
return image, image_to_canny
|
93 |
|
94 |
css="""
|
|
|
58 |
|
59 |
n_prompt = 'NSFW, nude, naked, porn, ugly'
|
60 |
|
61 |
+
|
62 |
|
63 |
# Canny preprocessing
|
64 |
image_to_canny = load_image(image_in)
|
|
|
89 |
height=h
|
90 |
).images[0]
|
91 |
|
92 |
+
image_redim, w, h = resize_image(image_in, "resized_input.jpg", 1024)
|
93 |
+
|
94 |
+
image = image.resize((w, h), Image.LANCZOS)
|
95 |
+
|
96 |
return image, image_to_canny
|
97 |
|
98 |
css="""
|