Update app.py

app.py

@@ -2,11 +2,12 @@ import torch
 from diffusers import DiffusionPipeline
 import gradio
 import os
+import spaces
 
 model_list = os.environ.get("MODELS").split(",")
 lora_list = os.environ.get("LORAS") # Not in use
 
-
+@spaces.GPU
 def generate(prompt, model)
     pipe = DiffusionPipeline.from_pretrained(
         model", 
@@ -38,3 +39,5 @@ with gr.Blocks() as demo:
 
     btn = gr.Button("Run")
     btn.click(fn=generate, inputs=[inp_prompt, inp_model], outputs=out)
+
+demo.launch()