Update app.py

app.py

@@ -52,12 +52,14 @@ def resize_image(input_path, output_path, target_height):
     # Save the resized image
     img.save(output_path)
 
-    return output_path
+    return output_path, new_width, target_height
 
 def infer(image_in, prompt, inference_steps, guidance_scale, control_weight):
     
     n_prompt = 'NSFW, nude, naked, porn, ugly'
 
+    image_in, w, h = resize_image(image_in, "resized_input.jpg", 1024)
+    
     # Canny preprocessing
     image_to_canny = load_image(image_in)
     image_to_canny = np.array(image_to_canny)
@@ -83,6 +85,8 @@ def infer(image_in, prompt, inference_steps, guidance_scale, control_weight):
         controlnet_conditioning=controlnet_conditioning,
         num_inference_steps=inference_steps,
         guidance_scale=guidance_scale,
+        width=w,
+        height=h
     ).images[0]
 
     return image, image_to_canny