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
@@ -53,6 +53,7 @@ def infer_canny(image_in, prompt, control_type, inference_steps, guidance_scale,
 
     if control_type == "canny":
         pipe = pipe_canny
+        pipe.to("cuda", torch.float16)
         # Canny preprocessing
         image_to_canny = load_image(image_in)
         image_to_canny = np.array(image_to_canny)
@@ -65,10 +66,9 @@ def infer_canny(image_in, prompt, control_type, inference_steps, guidance_scale,
     
     elif control_type == "tile":
         pipe = pipe_tile
+        pipe.to("cuda", torch.float16)
         control_image = load_image(image_in)
-
-   
-    
+ 
     # infer
     image = pipe(
         prompt=prompt,