fix: slow model initialization

frasaria/__init__.py

@@ -1,6 +1,18 @@
-from transformers import pipeline
+import os
+from dotenv import load_dotenv
 from parrot import Parrot
+from transformers import AutoModelForSequenceClassification, AutoTokenizer
+from sentence_transformers import SentenceTransformer
+import deepl
 
+load_dotenv()
+DEEPL_API_KEY = os.getenv('DEEPL_API_KEY')
+
+# initialize the models
 parrot = Parrot(model_tag='prithivida/parrot_paraphraser_on_T5')
-translator_id_to_en = pipeline('translation', model='PontifexMaximus/opus-mt-id-en-finetuned-id-to-en')
-translator_en_to_id = pipeline('translation', model='Helsinki-NLP/opus-mt-en-id')
+for model_tag in ['prithivida/parrot_fluency_model', 'prithivida/parrot_adequacy_model']:
+    AutoModelForSequenceClassification.from_pretrained(model_tag)
+    AutoTokenizer.from_pretrained(model_tag)
+SentenceTransformer("sentence-transformers/paraphrase-distilroberta-base-v2")
+
+translator = deepl.Translator(DEEPL_API_KEY)