Update app.py

app.py

@@ -67,7 +67,7 @@ pipe.scheduler = EulerAncestralDiscreteScheduler(
 # pipe.enable_freeu(b1=1.1, b2=1.1, s1=0.5, s2=0.7)
 pipe.enable_xformers_memory_efficient_attention()
 pipe.force_zeros_for_empty_prompt = False
-
+pipe.to('cuda')
 low_threshold = 100
 high_threshold = 200
 
@@ -94,7 +94,6 @@ def get_canny_filter(image):
 
 @spaces.GPU
 def generate_(prompt, negative_prompt, canny_image, num_steps, controlnet_conditioning_scale, seed):
-    pipe.to('cuda')
     generator = torch.Generator("cuda").manual_seed(seed)    
     images = pipe(
     prompt, negative_prompt=negative_prompt, image=canny_image, num_inference_steps=num_steps, controlnet_conditioning_scale=float(controlnet_conditioning_scale),