Update the model enpoint

app.py

@@ -8,8 +8,8 @@ from flores200_codes import flores_codes
 
 def load_models():
     # build model and tokenizer
-    model_name_dict = {'nllb-distilled-600M': 'facebook/nllb-200-distilled-600M',
-                  #'nllb-1.3B': 'facebook/nllb-200-1.3B',
+    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',
                   }
@@ -26,14 +26,6 @@ def load_models():
     return model_dict
 
 
-import os
-import torch
-import gradio as gr
-import time
-from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, pipeline
-from flores200_codes import flores_codes
-
-
 def load_models():
     # build model and tokenizer
     model_name_dict = {'nllb-distilled-600M': 'facebook/nllb-200-distilled-600M',
@@ -96,7 +88,7 @@ if __name__ == '__main__':
 
     outputs = gr.outputs.JSON()
 
-    title = "NLLB distilled 600M demo"
+    title = "NLLB distilled 1.3B demo"
 
     demo_status = "Demo is running on CPU"
     description = f"Details: https://github.com/facebookresearch/fairseq/tree/nllb. {demo_status}"