Update app.py

app.py

@@ -7,21 +7,17 @@ from diffusers import StableDiffusionControlNetPipeline, ControlNetModel, UniPCM
 
 # Load ControlNet model
 controlnet = ControlNetModel.from_pretrained(
-    "lllyasviel/sd-controlnet-canny", torch_dtype=torch.float16
+    "lllyasviel/sd-controlnet-canny", torch_dtype=torch.float32
 )
 
 # Load Stable Diffusion pipeline with ControlNet
 pipe = StableDiffusionControlNetPipeline.from_pretrained(
-    "runwayml/stable-diffusion-v1-5", controlnet=controlnet, torch_dtype=torch.float16
+    "runwayml/stable-diffusion-v1-5", controlnet=controlnet, torch_dtype=torch.float32
 )
 
 # Set the scheduler
 pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
 
-# Enable optimization for faster generation
-pipe.enable_model_cpu_offload()
-pipe.enable_xformers_memory_efficient_attention()
-
 def process_and_generate(image, prompt, num_inference_steps, guidance_scale):
     # Convert PIL image to numpy array
     image = np.array(image)