Update app.py

app.py

@@ -11,8 +11,8 @@ def load_models():
     model_name_dict = {
                   #'nllb-distilled-600M': 'facebook/nllb-200-distilled-600M',
                   #'nllb-1.3B': 'facebook/nllb-200-1.3B',
-                  #'nllb-distilled-1.3B': 'facebook/nllb-200-distilled-1.3B',
-                  'nllb-3.3B': 'facebook/nllb-200-3.3B',
+                  'nllb-distilled-1.3B': 'facebook/nllb-200-distilled-1.3B',
+                  #'nllb-3.3B': 'facebook/nllb-200-3.3B',
                   }
 
     model_dict = {}
@@ -29,7 +29,7 @@ def load_models():
 
 def translation(source, target, text):
     if len(model_dict) == 2:
-        model_name = 'nllb-3.3B'
+        model_name = 'nllb-distilled-1.3B'
 
     start_time = time.time()
     source = flores_codes[source]