Update app.py

app.py

@@ -3,7 +3,7 @@ import cv2
 import numpy as np
 from loadimg import load_img
 import gradio as gr
-
+import spaces
 
 model = AnimeSegmentation.from_pretrained("skytnt/anime-seg") 
 
@@ -36,7 +36,7 @@ def get_mask(model, input_img, use_amp=True, s=640):
         pred = cv2.resize(pred, (w0, h0))[:, :, np.newaxis]
         return pred
 
-
+@spaces.GPU
 def process(img):
     path = load_img(img,output_type="str")
     img = cv2.cvtColor(cv2.imread(path, cv2.IMREAD_COLOR), cv2.COLOR_BGR2RGB)