Update func

resources.py

@@ -39,9 +39,10 @@ def load_model_and_tokenizer(device="cpu"):
     return encoder, tokenizer
 
 
-def get_gtr_embeddings(text_list,
+def get_gtr_embeddings(text_list: list[str],
                        encoder: PreTrainedModel,
-                       tokenizer: PreTrainedTokenizer) -> torch.Tensor:
+                       tokenizer: PreTrainedTokenizer,
+                       ) -> torch.Tensor:
 
     inputs = tokenizer(text_list,
                        return_tensors="pt",