Update app.py

app.py

@@ -5,12 +5,8 @@ import time
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, pipeline
 
 
-def load_models():
-    model = AutoModelForSeq2SeqLM.from_pretrained("facebook/nllb-200-distilled-1.3B")
-    tokenizer = AutoTokenizer.from_pretrained("facebook/nllb-200-distilled-1.3B")
-    
-    return model, tokenizer
-
+model = AutoModelForSeq2SeqLM.from_pretrained("facebook/nllb-200-distilled-1.3B")
+tokenizer = AutoTokenizer.from_pretrained("facebook/nllb-200-distilled-1.3B")
 
 def translation(source, target, text):
 
@@ -24,11 +20,6 @@ def translation(source, target, text):
 
 
 if __name__ == '__main__':
-    print('\tinit models')
- 
-    model, tokenizer = load_models()
-    global model
-    global tokenizer
     
     # define gradio demo
     lang_codes = ["eng_Latn", "fuv_Latn", "fra_Latn", "arb_Arab"]