Update app.py

app.py

@@ -1,6 +1,7 @@
 import os
 import gradio as gr
 import json
+import spaces
 import logging
 import torch
 from PIL import Image
@@ -98,6 +99,7 @@ def load_fast_model():
     return pipe, config
 
 # Generate image
+@spaces.GPU
 def generate_image(prompt, resolution, seed, guidance_scale, num_inference_steps):
     global pipe, MODEL_CONFIG
     if pipe is None: