Zero GPU ready

app.py

@@ -1,4 +1,5 @@
 import gradio as gr
+import spaces
 import os
 import sys
 import subprocess
@@ -54,11 +55,10 @@ def resize_image(input_path, output_path, target_height):
 
     return output_path, new_width, target_height
 
+@spaces.GPU()
 def infer(image_in, prompt, inference_steps, guidance_scale, control_weight, progress=gr.Progress(track_tqdm=True)):
     
     n_prompt = 'NSFW, nude, naked, porn, ugly'
-
-    
     
     # Canny preprocessing
     image_to_canny = load_image(image_in)