update

app.py

@@ -6,6 +6,7 @@ import torch.nn.functional as F
 from models.arch.RDnet_ import FullNet_NLP
 from models.arch.classifier import PretrainedConvNext
 import torchvision.transforms.functional as TF
+import spaces
 print(f"Is CUDA available: {torch.cuda.is_available()}")
 # True
 print(f"CUDA device: {torch.cuda.get_device_name(torch.cuda.current_device())}")
@@ -26,6 +27,7 @@ def align(x1):
     h, w = h // 32 * 32, w // 32 * 32
     x1 = np.resize(x1, (h, w, x1.shape[2]))
     return x1
+@spaces.GPU(duration=120)
 def predict(img):
     with torch.no_grad():
         img=align(img)