Update app.py

app.py

@@ -92,6 +92,17 @@ def get_canny_filter(image):
     canny_image = Image.fromarray(image)
     return canny_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),
+    generator=generator,
+    ).images
+    return images
+
+
 def process(input_image, prompt, negative_prompt, num_steps, controlnet_conditioning_scale, seed):
     
     # resize input_image to 1024x1024
@@ -103,15 +114,7 @@ def process(input_image, prompt, negative_prompt, num_steps, controlnet_conditio
 
     return [canny_image,images[0]]
 
-@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),
-    generator=generator,
-    ).images
-    return images
+
     
 block = gr.Blocks().queue()