Update app.py

app.py

@@ -289,8 +289,26 @@ model.eval()
 bot_name = "WeASK"
 
 ###removed
+from transformers import MBartForConditionalGeneration, MBart50Tokenizer
 
+def download_model():
+    model_name = "facebook/mbart-large-50-many-to-many-mmt"
+    model = MBartForConditionalGeneration.from_pretrained(model_name)
+    tokenizer = MBart50Tokenizer.from_pretrained(model_name)
+    return model, tokenizer
+
+model, tokenizer = download_model()
+
+
+
+################################
 def get_response(input_text):
+    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)
+
+
+
     #print("Let's chat! (type 'quit' to exit)")
     #while True:
     # sentence = "do you use credit cards?"
@@ -303,7 +321,7 @@ def get_response(input_text):
     #if sentence== "quit":
         #break
 
-    sentence= tokenize(input_text)
+    sentence= tokenize(translation)
     X = bag_of_words(sentence, all_words)
     X = X.reshape(1, X.shape[0])
     X = torch.from_numpy(X).to(device)