kovacsvi commited on
Commit
79bbcba
·
1 Parent(s): 5de5900
Files changed (1) hide show
  1. interfaces/cap_minor_media.py +1 -1
interfaces/cap_minor_media.py CHANGED
@@ -57,7 +57,7 @@ def check_huggingface_path(checkpoint_path: str):
57
  def build_huggingface_path(language: str, domain: str):
58
  return ("poltextlab/xlm-roberta-large-pooled-cap-media", "poltextlab/xlm-roberta-large-pooled-cap-minor-v3")
59
 
60
- @spaces.GPU
61
  def predict(text, major_model_id, minor_model_id, tokenizer_id, HF_TOKEN=None):
62
  device = torch.device("cuda:0")
63
 
 
57
  def build_huggingface_path(language: str, domain: str):
58
  return ("poltextlab/xlm-roberta-large-pooled-cap-media", "poltextlab/xlm-roberta-large-pooled-cap-minor-v3")
59
 
60
+ @spaces.GPU(duration=30)
61
  def predict(text, major_model_id, minor_model_id, tokenizer_id, HF_TOKEN=None):
62
  device = torch.device("cuda:0")
63