Update app.py

app.py

@@ -292,9 +292,7 @@ bot_name = "WeASK"
 from transformers import MBartForConditionalGeneration, MBart50TokenizerFast
 
 #def download_model():
-model_name = "facebook/mbart-large-50-many-to-many-mmt"
-model = MBartForConditionalGeneration.from_pretrained(model_name)
-tokenizer = MBart50TokenizerFast.from_pretrained(model_name)
+
 
 #model, tokenizer = download_model()
 
@@ -302,12 +300,16 @@ tokenizer = MBart50TokenizerFast.from_pretrained(model_name)
 
 ################################
 def get_response(input_text):
+    model_name = "facebook/mbart-large-50-many-to-many-mmt"
+    model = MBartForConditionalGeneration.from_pretrained(model_name)
+    tokenizer = MBart50TokenizerFast.from_pretrained(model_name)
+
     model_inputs = tokenizer(input_text, return_tensors="pt")
     generated_tokens = model.generate(**model_inputs,forced_bos_token_id=tokenizer.lang_code_to_id["en_XX"])
     translation = tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)
     #list1=["Welcome","To","Tutorials","Point"]
     #string1="".join(map(str,list1))
-    string2=" ".join(map(str,translation))
+    string2=" ".join(map(str,translation ))
     #print(string2)