zeroGPU for cap task

interfaces/cap.py

@@ -82,6 +82,7 @@ def build_huggingface_path(language: str, domain: str):
     else:
         return "poltextlab/xlm-roberta-large-pooled-cap"
 
+@spaces.GPU(duration=5)
 def predict(text, model_id, tokenizer_id):
     device = torch.device("cpu")
     model = AutoModelForSequenceClassification.from_pretrained(model_id, low_cpu_mem_usage=True, device_map="auto", offload_folder="offload", token=HF_TOKEN)