Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -26,7 +26,9 @@ def main():
|
|
| 26 |
|
| 27 |
# Left column: Radio buttons for summarizer choice
|
| 28 |
with col1:
|
| 29 |
-
radio_options = ['Abstractive
|
|
|
|
|
|
|
| 30 |
help_text = "Abstractive: Abstractive summarization generates a summary that may contain words not present in the original text. " \
|
| 31 |
"It uses a fine-tuned model on BART-large-CNN.<br>" \
|
| 32 |
"Extractive: Extractive summarization selects and extracts sentences or phrases directly from the original text to create a summary using the TextRank algorithm.<br>" \
|
|
@@ -66,7 +68,7 @@ def main():
|
|
| 66 |
|
| 67 |
# Perform extractive summarization
|
| 68 |
if radio_selection == "Abstractive (Fine-tuned BART)":
|
| 69 |
-
summary =
|
| 70 |
st.session_state.summary = summary
|
| 71 |
|
| 72 |
# Perform extractive summarization
|
|
|
|
| 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)", 'Keyword Extraction', 'Keyphrase Extraction']
|
| 31 |
+
|
| 32 |
help_text = "Abstractive: Abstractive summarization generates a summary that may contain words not present in the original text. " \
|
| 33 |
"It uses a fine-tuned model on BART-large-CNN.<br>" \
|
| 34 |
"Extractive: Extractive summarization selects and extracts sentences or phrases directly from the original text to create a summary using the TextRank algorithm.<br>" \
|
|
|
|
| 68 |
|
| 69 |
# Perform extractive summarization
|
| 70 |
if radio_selection == "Abstractive (Fine-tuned BART)":
|
| 71 |
+
summary = summarize_with_bart_ft(file_content)
|
| 72 |
st.session_state.summary = summary
|
| 73 |
|
| 74 |
# Perform extractive summarization
|