Update utils.py

utils.py

@@ -10,7 +10,6 @@ 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]:
@@ -58,7 +57,6 @@ 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.