Update app.py

app.py

@@ -8,7 +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',
+    model_name_dict = { 'nllb-3.3B': 'facebook/nllb-200-3.3B',
+                  #'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',
@@ -28,7 +29,8 @@ def load_models():
 
 def translation(source, target, text):
     if len(model_dict) == 2:
-        model_name = 'nllb-distilled-600M'
+        # model_name = 'nllb-distilled-600M'
+        model_name = 'nllb-3.3B'
 
     start_time = time.time()
     source = flores_codes[source]
@@ -61,18 +63,18 @@ if __name__ == '__main__':
     lang_codes = list(flores_codes.keys())
     #inputs = [gr.inputs.Radio(['nllb-distilled-600M', 'nllb-1.3B', 'nllb-distilled-1.3B'], label='NLLB Model'),
     inputs = [gr.inputs.Dropdown(lang_codes, default='English', label='Source'),
-              gr.inputs.Dropdown(lang_codes, default='Korean', label='Target'),
+              gr.inputs.Dropdown(lang_codes, default='Shan', label='Target'),
               gr.inputs.Textbox(lines=5, label="Input text"),
               ]
 
     outputs = gr.outputs.JSON()
 
-    title = "NLLB distilled 600M demo"
+    title = "NLLB Translation demo"
 
     demo_status = "Demo is running on CPU"
     description = f"Details: https://github.com/facebookresearch/fairseq/tree/nllb. {demo_status}"
     examples = [
-    ['English', 'Korean', 'Hi. nice to meet you']
+    ['English', 'Shan', 'Hi. nice to meet you']
     ]
 
     gr.Interface(translation,