EE21 commited on
Commit
da1f55e
1 Parent(s): c0ade28

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -65,7 +65,12 @@ def main():
65
  st.session_state.summary = summary
66
 
67
  # Perform extractive summarization
68
- if radio_selection == "Abstractive":
 
 
 
 
 
69
  summary = summarize_with_bart(file_content)
70
  st.session_state.summary = summary
71
 
 
65
  st.session_state.summary = summary
66
 
67
  # Perform extractive summarization
68
+ if radio_selection == "Abstractive (Fine-tuned BART)":
69
+ summary = summarize_with_bart(file_content)
70
+ st.session_state.summary = summary
71
+
72
+ # Perform extractive summarization
73
+ if radio_selection == "Abstractive (BART-large-CNN)":
74
  summary = summarize_with_bart(file_content)
75
  st.session_state.summary = summary
76