pull model files

ar_en

@@ -1 +0,0 @@
-Subproject commit b5626c21d9814e83302354362e60d813003f8b97

en_ar

@@ -1 +0,0 @@
-Subproject commit 76d2a612d5c6b4cc8fe16bd7608e7a0809a96ba5

translate.py

@@ -13,13 +13,13 @@ formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(messag
 file_handler.setFormatter(formatter)
 logger.addHandler(file_handler)
 
-model_to_ar = MarianMTModel.from_pretrained("./en_ar/", output_attentions=True)
-model_from_ar = MarianMTModel.from_pretrained("./ar_en/", output_attentions=True)
+model_to_ar = MarianMTModel.from_pretrained("guymorlan/levanti_translate_en_ar", output_attentions=True)
+model_from_ar = MarianMTModel.from_pretrained("guymorlan/levanti_translate_ar_en", output_attentions=True)
 model_to_ar_ct2 = ctranslate2.Translator("./en_ar_ct2/")
 model_from_ar_ct2 = ctranslate2.Translator("./ar_en_ct2/")
 
-tokenizer_to_ar = AutoTokenizer.from_pretrained("./en_ar/")
-tokenizer_from_ar = AutoTokenizer.from_pretrained("./ar_en/")
+tokenizer_to_ar = AutoTokenizer.from_pretrained("guymorlan/levanti_translate_en_ar")
+tokenizer_from_ar = AutoTokenizer.from_pretrained("guymorlan/levanti_translate_ar_en")
 print("Done loading models")
 
 dialect_map = {
@@ -98,4 +98,4 @@ def run_translate(text, dialect=None):
 
         text = f"{dialect} {text}" if dialect else text
         return translate(text, model_to_ar_ct2, model_to_ar, tokenizer_to_ar,
-                          to_arabic=True, threshold=None, layer=2, head=7)
+                          to_arabic=True, threshold=None, layer=2, head=7)