Update app.py

app.py

@@ -5,7 +5,9 @@ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, pipeline
 from flores200_codes import flores_codes
 
 def load_models():
-    model_name_dict = {'nllb-distilled-600M': 'facebook/nllb-200-distilled-600M'}
+    model_name_dict = {'nllb-distilled-1.3B': 'facebook/nllb-200-distilled-1.3B',
+                       #'nllb-distilled-600M': 'facebook/nllb-200-distilled-600M'
+                      }
     model_dict = {}
     for call_name, real_name in model_name_dict.items():
         print(f'\tLoading model: {call_name}')
@@ -20,7 +22,7 @@ model_dict = load_models()
 
 def translate_text(source_lang, target_lang, input_text):
     if len(model_dict) == 2:
-        model_name = 'nllb-distilled-600M'
+        model_name = 'nllb-distilled-1.3B'
     start_time = time.time()
     source = flores_codes.get(source_lang)
     target = flores_codes.get(target_lang)