Update app.py

app.py

@@ -3,6 +3,13 @@ from transformers import pipeline
 import textwrap
 
 st.title('Hugging Face BERT Summarizer')
+
+# List of models
+models = ["sshleifer/distilbart-cnn-12-6", "facebook/bart-large-cnn", "t5-base", "t5-large", "google/pegasus-newsroom"]
+
+# Dropdown model selector
+model = st.sidebar.selectbox("Choose a model", models)
+
 uploaded_file = st.file_uploader("Choose a .txt file", type="txt")
 
 # Add sliders to the sidebar
@@ -13,7 +20,7 @@ if uploaded_file is not None:
     user_input = uploaded_file.read().decode('utf-8')
 
     if st.button('Summarize'):
-        summarizer = pipeline('summarization')
+        summarizer = pipeline('summarization', model=model)
         summarized_text = ""
 
         # Split the text into chunks of approximately 500 words each