Update utils.py

utils.py

@@ -10,6 +10,7 @@ from torch.quantization import quantize_dynamic
 from torch import nn, qint8
 from transformers import MBartForConditionalGeneration, AutoConfig
 
+import spaces
 
 @st.cache_resource(show_spinner=False)
 def get_resources(multilingual: bool, src_lang: str, quantize: bool = True, no_cuda: bool = False) -> Tuple[MBartForConditionalGeneration, AMRTokenizerWrapper]:
@@ -57,6 +58,7 @@ def get_resources(multilingual: bool, src_lang: str, quantize: bool = True, no_c
     return model, tok_wrapper
 
 
+@spaces.GPU
 def translate(texts: List[str], src_lang: str, model: MBartForConditionalGeneration, tok_wrapper: AMRTokenizerWrapper, **gen_kwargs) -> Dict[str, List[Union[penman.Graph, ParsedStatus]]]:
     """Translates a given text of a given source language with a given model and tokenizer. The generation is guided by
     potential keyword-arguments, which can include arguments such as max length, logits processors, etc.