add zeroGPU support

app.py

@@ -2,6 +2,7 @@ import random
 
 import gradio as gr
 import numpy as np
+import spaces
 import torch
 import torchvision.transforms as transforms
 from PIL import Image
@@ -33,6 +34,7 @@ MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 1024
 
 
+@spaces.GPU
 def infer(
     input_image,
     prompt1,

requirements.txt

@@ -10,4 +10,5 @@ DeepCache
 pandas
 numpy
 torchmetrics[image]
-gradio
+gradio
+spaces