Update app.py

app.py

@@ -294,9 +294,9 @@ model = MBartForConditionalGeneration.from_pretrained(model_name)
 tokenizer = MBart50Tokenizer.from_pretrained(model_name)
 
 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)
+    #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)
     
     #string2=" ".join(map(str,translation ))
 
@@ -315,7 +315,7 @@ def get_response(input_text):
     #if sentence== "quit":
         #break
 
-    sentence= tokenize(translation)
+    sentence= tokenize(input_text)
     X = bag_of_words(sentence, all_words)
     X = X.reshape(1, X.shape[0])
     X = torch.from_numpy(X).to(device)