Update

app.py

@@ -8,6 +8,7 @@ import sys
 import gradio as gr
 import numpy as np
 import PIL.Image
+import spaces
 import torch
 import torchvision.transforms as T
 from huggingface_hub import hf_hub_download
@@ -37,6 +38,7 @@ model.eval()
 model.to(device)
 
 
+@spaces.GPU
 @torch.inference_mode()
 def predict(image: PIL.Image.Image) -> np.ndarray:
     data = transform(image)

requirements.txt

@@ -1,5 +1,6 @@
 numpy==1.26.4
 opencv-python-headless==4.9.0.80
 Pillow==10.3.0
+spaces==0.28.3
 torch==2.0.1
 torchvision==0.15.2