Update

README.md

@@ -1,5 +1,5 @@
 ---
-title: MedChat
+title: MedChatRe
 emoji: 🐠
 colorFrom: purple
 colorTo: green

app.py

@@ -8,15 +8,15 @@ from transformers import AutoTokenizer
 from langdetect import detect
 import subprocess
 
-tokenizer = AutoTokenizer.from_pretrained("facebook/bart-base")
+tokenizer = AutoTokenizer.from_pretrained("GuysTrans/bart-base-re-attention-seq-512")
 
-vn_tokenizer = AutoTokenizer.from_pretrained("GuysTrans/bart-base-vn-ehealth-vn-tokenizer")
+vn_tokenizer = AutoTokenizer.from_pretrained("GuysTrans/bart-base-vn-re-attention-vn-tokenizer")
 
 model = AutoModelForSeq2SeqLM.from_pretrained(
-    "GuysTrans/bart-base-finetuned-xsum", revision="worked")
+    "GuysTrans/bart-base-re-attention-seq-512")
 
 vn_model = AutoModelForSeq2SeqLM.from_pretrained(
-    "GuysTrans/bart-base-vn-ehealth-vn-tokenizer", revision="worked")
+    "GuysTrans/bart-base-vn-re-attention-vn-tokenizer")
 
 map_words = {
     "Hello and Welcome to 'Ask A Doctor' service": "",