book-summary / app.py
bqpxben's picture
Upload 2 files
d3b8ec3
raw
history blame
No virus
525 Bytes
import streamlit as st
from transformers import pipeline
generate_summary = pipeline("book-summary-tools", model=" bqpxben/book-summary-tools")
def main():
st.title("Book Summarization App")
# Get user input
user_input = st.text_area("Enter the text you want to summarize:", "")
if st.button("Summarize"):
# Perform summarization
summary = generate_summary(user_input)
# Display result
st.write("Summary:")
st.write(summary)
if __name__ == "__main__":
main()