fix: space gpu

app.py

@@ -8,11 +8,13 @@ import torch.nn.functional as F
 import os
 import imquality.brisque as brisque
 from loss.niqe_utils import *
+import spaces
 
 eval_net = CIDNet().cuda()
 eval_net.trans.gated = True
 eval_net.trans.gated2 = True
 
+@spaces.GPU(duration=120)
 def process_image(input_img,score,model_path,gamma=1.0,alpha_s=1.0,alpha_i=1.0):
     if model_path is None:
         return input_img,"Please choose a model weights."

requirements.txt

@@ -8,3 +8,4 @@ torchvision
 einops
 opencv-python
 gradio
+spaces