add model for zhaiyao

app.py

@@ -5,6 +5,7 @@ import os
 import time
 import torch
 from config import Config
+from ebart import PegasusSummarizer
 from transformers import BertConfig, BertTokenizer, BertForSequenceClassification
 
 
@@ -28,6 +29,8 @@ def greet(inputStr):
     )
     model.to(config.device)
 
+    summarizerModel = PegasusSummarizer()
+    inputStr = summarizerModel.generate_summary(inputStr, 200, 64)    
 
     model.load_state_dict(torch.load(config.saved_model))
     model.eval()