use zero gpus

edit_app.py

@@ -1,17 +1,20 @@
 from __future__ import annotations
 
-import math
-from glob import glob
 from functools import partial
+from glob import glob
+import math
 import random
+from PIL import Image, ImageOps
 
 import gradio as gr
 import torch
-from PIL import Image, ImageOps
+
 from datasets import load_dataset
 from diffusers import StableDiffusionInstructPix2PixPipeline, EulerAncestralDiscreteScheduler
+import spaces
 
 
+@spaces.GPU(duration=120)
 def generate(
     input_image: Image.Image,
     instruction: str,