add zerogpu support

app.py

@@ -38,6 +38,7 @@ import torch
 
 import gradio as gr
 from huggingface_hub import snapshot_download
+import spaces
 
 from model import StableMultiDiffusionSDXLPipeline
 from util import seed_everything
@@ -302,6 +303,7 @@ def import_state(state, json_text):
 
 ### Main worker
 
+@spaces.GPU
 def generate(state, *args, **kwargs):
     return models[state.model_id](*args, **kwargs)