Update app.py

app.py

@@ -1,6 +1,7 @@
 from diffusers import StableDiffusionPipeline, StableDiffusionImg2ImgPipeline, UniPCMultistepScheduler
 import gradio as gr
 import torch
+import spaces
 from PIL import Image
 
 model_id = 'Norod78/sd2-simpsons-blip'
@@ -26,6 +27,7 @@ def error_str(error, title="Error"):
     return f"""#### {title}
             {error}"""  if error else ""
 
+@spaces.GPU
 def inference(prompt, guidance, steps, width=512, height=512, seed=0, img=None, strength=0.5, neg_prompt=""):
 
   if torch.cuda.is_available():