Update app.py

app.py

@@ -1,4 +1,5 @@
 import gradio as gr
++import spaces
 
 #gr.load("models/sergon19/green_bg_LoRa10").launch()
 
@@ -16,6 +17,7 @@ pipe = DiffusionPipeline.from_pretrained(
 pipe.load_lora_weights("sergon19/green_bg_LoRa10")
 _ = pipe.to("cuda")
 
++@spaces.GPU
 def generate_image(prompt):
     image = pipe(prompt=prompt, num_inference_steps=25).images[0]
     return image