Update app.py

app.py

@@ -71,36 +71,34 @@ def translation(source, target, text):
 
 
 
-if __name__ == '__main__':
-    print('\tinit models')
 
-    global model_dict
 
-    model_dict = load_models()
+global model_dict
+
+model_dict = load_models()
     
     # define gradio demo
-    lang_codes = list(flores_codes.keys())
+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.Textbox(lines=5, label="Input text"),
-              ]
-
-    outputs = gr.outputs.JSON()
-
-    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}"
-    examples = [
-    ['English', 'Urdu', 'Hi. nice to meet you']
-    ]
-
-    gr.Interface(translation,
-                 inputs=[gr.Textbox(title="source"),gr.Textbox(title="target"),gr.Textbox(title="text")],
-                 outputs=gr.Textbox(),
-                 title=title,
-                 description=description,
-                 ).launch()
+
+
+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}"
+examples = [
+['English', 'Urdu', 'Hi. nice to meet you']
+]
+
+gr.Interface(translation,
+            inputs = [gr.inputs.Dropdown(lang_codes, default='English', label='Source'),
+                      gr.inputs.Dropdown(lang_codes, default='Korean', label='Target'),
+                      gr.inputs.Textbox(lines=5, label="Input text"),
+                      ],
+
+            outputs = gr.outputs.JSON(),
+             title=title,
+             description=description,
+             ).launch()