Update app.py

app.py

@@ -5,7 +5,7 @@ from torch.utils.data import Dataset
 import torch.utils.data
 from models import *
 from utils import *
-st.title("UniLM chatbot")
+st.title("UniLM Beta Testing")
 st.subheader("AI language chatbot by Webraft-AI")
 #Picking what NLP task you want to do
 
@@ -14,7 +14,7 @@ st.subheader("Start the conversation")
 text2 = st.text_input('Human: ') #text is stored in this variable
 
 load_checkpoint = True
-ckpt_path = 'checkpoint_9.pth.tar'
+ckpt_path = 'checkpoint_190.pth.tar'
 with open('WORDMAP_corpus.json', 'r') as j:
     word_map = json.load(j)
 
@@ -65,7 +65,7 @@ def remove_punc(string):
     return no_punct.lower()
 question = remove_punc(text2)
 
-max_len = 128
+max_len = 153
 enc_qus = [word_map.get(word, word_map['<unk>']) for word in question.split()]
 question = torch.LongTensor(enc_qus).to(device).unsqueeze(0)
 question_mask = (question!=0).to(device).unsqueeze(1).unsqueeze(1)