Update app.py

app.py

@@ -20,12 +20,12 @@ controlnet_tile = SD3ControlNetModel.from_pretrained("InstantX/SD3-Controlnet-Ti
 pipe_canny = StableDiffusion3ControlNetPipeline.from_pretrained(
             "stabilityai/stable-diffusion-3-medium-diffusers",
             controlnet=controlnet_canny
-        )
+        ).to("cuda", torch.float16)
 
 pipe_tile = StableDiffusion3ControlNetPipeline.from_pretrained(
             "stabilityai/stable-diffusion-3-medium-diffusers",
             controlnet=controlnet_tile
-        )
+        ).to("cuda", torch.float16)
 
 def resize_image(input_path, output_path, target_height):
     # Open the input image
@@ -67,7 +67,7 @@ def infer_canny(image_in, prompt, control_type, inference_steps, guidance_scale,
         pipe = pipe_tile
         control_image = load_image(image_in)
 
-   pipe.to("cuda", torch.float16)
+   
     
     # infer
     image = pipe(