Update app.py

app.py

@@ -19,6 +19,11 @@ model_org = load_model(cp_org)
 model_aug  = load_model(cp_aug)
 tokenizer = load_tokenizer("VietAI/vit5-base")
 
+def processed(text):
+    processed_text = text.replace('\n', ' ')
+    processed_text = processed_text.lower()
+    return processed_text
+    
 def summarize(text, model, tokenizer, num_beams=4, device='cpu'):
     model.eval()
     model.to(device)
@@ -30,8 +35,9 @@ def summarize(text, model, tokenizer, num_beams=4, device='cpu'):
     
 text = st.text_area('Nhập tiêu đề vào đây')
 if text:
-    re1 = summarize(model_org, tokenizer, text)
-    re2 = summarize(model_aug, tokenizer, text)
+    processed_text = processed(text)
+    re1 = summarize(processed_text, model_org, tokenizer)
+    re2 = summarize(processed_text, model_aug, tokenizer)
     out = {
         'Result from model with original data': re1,
         'Result from model with augmented data': re2