Add vn lang

app.py

@@ -4,6 +4,7 @@ import os
 import re
 from transformers import AutoModelForSeq2SeqLM
 from transformers import AutoTokenizer
+from langdetect import detect
 import subprocess
 
 tokenizer = AutoTokenizer.from_pretrained("facebook/bart-base")
@@ -11,6 +12,9 @@ tokenizer = AutoTokenizer.from_pretrained("facebook/bart-base")
 model = AutoModelForSeq2SeqLM.from_pretrained(
     "GuysTrans/bart-base-finetuned-xsum")
 
+vn_model = AutoModelForSeq2SeqLM.from_pretrained(
+    "GuysTrans/bart-base-vn-ehealth")
+
 map_words = {
     "Hello and Welcome to 'Ask A Doctor' service": "",
     "Hello,": "",
@@ -53,7 +57,11 @@ PAGE_ACCESS_TOKEN = os.environ['PAGE_ACCESS_TOKEN']
 
 
 def get_bot_response(message):
-    return post_process(generate_summary(message, model)[1][0])
+    lang = detect(message)
+    model_use = model
+    if lang == "vn":
+        model_use = vn_model
+    return post_process(generate_summary(message, model_use)[1][0])
 
 
 def verify_webhook(req):

requirements.txt

@@ -1,3 +1,4 @@
 transformers
 flask
 torch
+langdetect