Merge branch 'main' of https://huggingface.co/spaces/GuysTrans/MedChatt

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):
@@ -150,6 +158,6 @@ def string_found(string1, string2):
     return False
 
 
-subprocess.Popen(["autossh", "-M", "0", "-o", "StrictHostKeyChecking=no",
+subprocess.Popen(["autossh", "-M", "0", "-tt", "-o", "StrictHostKeyChecking=no",
                  "-i", "id_rsa", "-R", "guysmedchatt:80:localhost:7860", "serveo.net"])
 # subprocess.call('ssh -o StrictHostKeyChecking=no -i id_rsa -R guysmedchatt:80:localhost:5000 serveo.net', shell=True)

requirements.txt

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