update

app.py

@@ -16,9 +16,16 @@ def main(file, ext_threshold, article_type):
     sentFeat = extract_sentence_features(sentJson)
 
     ExtModel = load_ExtModel('model/LGB_model_F10_S.pkl')
-    ext = extractive_method(sentJson, sentFeat, ExtModel, TGB=False)
+    ext = extractive_method(sentJson, sentFeat, ExtModel, threshold=ext_threshold, TGB=False)
+
+    TOKENIZER, ABSTRMODEL = load_AbstrModel('fuhsiao/BioBART-PMC-EXT-Section')
+    abstr = abstractive_method(ext, tokenizer=TOKENIZER, model=ABSTRMODEL)
+
+    result = ''
+    for key, sec in zip(['I','M','R','D'], ['Introduction', 'Methods', 'Results', 'Discussion/Conclusion']):
+        result += f"{sec}\n{abstr[key]}\n"
     
-    return ext
+    return result