EE21 commited on
Commit
b3bc472
1 Parent(s): f25eb0c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -1
app.py CHANGED
@@ -26,7 +26,7 @@ def main():
26
 
27
  # Left column: Radio buttons for summarizer choice
28
  with col1:
29
- radio_options = ['Abstractive (Fine-tuned BART)', "Abstractive (BART-large-CNN)", 'Extractive (TextRank)',
30
  "Extractive (Latent Semantic Analysis)", 'Keyphrase Extraction (RAKE)', 'Keyword Extraction (RAKE)']
31
 
32
  help_text = "Abstractive: Abstractive summarization generates a summary that may contain words not present in the original text. " \
@@ -76,6 +76,16 @@ def main():
76
  summary = summarize_with_bart(file_content)
77
  st.session_state.summary = summary
78
 
 
 
 
 
 
 
 
 
 
 
79
  # Perform Keyword Extraction
80
  if radio_selection == "Keyword Extraction (RAKE)":
81
  summary = extract_keywords(file_content)
 
26
 
27
  # Left column: Radio buttons for summarizer choice
28
  with col1:
29
+ radio_options = ["Abstractive (T5)", "Abstractive (LED)", 'Abstractive (Fine-tuned BART)', "Abstractive (BART-large-CNN)", 'Extractive (TextRank)',
30
  "Extractive (Latent Semantic Analysis)", 'Keyphrase Extraction (RAKE)', 'Keyword Extraction (RAKE)']
31
 
32
  help_text = "Abstractive: Abstractive summarization generates a summary that may contain words not present in the original text. " \
 
76
  summary = summarize_with_bart(file_content)
77
  st.session_state.summary = summary
78
 
79
+ # Perform extractive summarization
80
+ if radio_selection == "Abstractive (T5)":
81
+ summary = summarize_with_t5(file_content)
82
+ st.session_state.summary = summary
83
+
84
+ # Perform extractive summarization
85
+ if radio_selection == "Abstractive (LED))":
86
+ summary = summarize_with_led(file_content)
87
+ st.session_state.summary = summary
88
+
89
  # Perform Keyword Extraction
90
  if radio_selection == "Keyword Extraction (RAKE)":
91
  summary = extract_keywords(file_content)