Update app.py

app.py

@@ -10,6 +10,8 @@ import subprocess
 
 tokenizer = AutoTokenizer.from_pretrained("facebook/bart-base")
 
+vn_tokenizer = AutoTokenizer.from_pretrained("GuysTrans/bart-base-vn-ehealth-vn-tokenizer")
+
 model = AutoModelForSeq2SeqLM.from_pretrained(
     "GuysTrans/bart-base-finetuned-xsum", revision="worked")
 
@@ -37,7 +39,7 @@ word_remove_sentence = [
 ]
 
 
-def generate_summary(question, model):
+def generate_summary(question, model, tokenizer):
     inputs = tokenizer(
         question,
         padding="max_length",
@@ -64,11 +66,13 @@ PAGE_ACCESS_TOKEN = os.environ['PAGE_ACCESS_TOKEN']
 def get_bot_response(message):
     lang = detect(message)
     model_use = model
+    tokenizer_use = tokenizer
     template = "Welcome to MedForRum chatbot service. %s. Thanks for asking on MedForum."
     if lang == "vi":
         model_use = vn_model
+        tokenizer_use = vn_tokenizer
         template = "Chào mừng bạn đến với dịch vụ MedForRum chatbot. %s. Cảm ơn bạn đã sử dụng MedForum."
-    return template % post_process(generate_summary(message, model_use)[1][0])
+    return template % post_process(generate_summary(message, model_use, tokenizer_use)[1][0])
 
 
 def verify_webhook(req):