EE21 commited on
Commit
1918f01
1 Parent(s): 2f8382c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
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', 'Extractive (TextRank)', "Extractive (Latent Semantic Analysis)", 'Keyword Extraction', 'Keyphrase Extraction']
 
 
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 = summarize_with_bart(file_content)
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