Spaces:
Configuration error
Configuration error
File size: 1,076 Bytes
699a866 b56f8fb 0478907 b56f8fb 0478907 b56f8fb |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 |
import streamlit as st
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
import torch
# Model name
MODEL_NAME = "AbdullahAlnemr1/flan-t5-summarizer"
# Load tokenizer and model
tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
model = AutoModelForSeq2SeqLM.from_pretrained(MODEL_NAME)
st.title("Text Summarizer(Encoder-Decoder)")
input_text = st.text_area("Enter text to summarize:", height=200)
max_new_tokens = st.slider("Max summary length (tokens)", min_value=20, max_value=200, value=100)
if st.button("Generate Summary"):
if input_text.strip() == "":
st.warning("Enter text to summarize")
else:
# Tokenize input
inputs = tokenizer(input_text, return_tensors="pt", truncation=True)
# Generate summary
outputs = model.generate(
inputs["input_ids"],
max_new_tokens=max_new_tokens,
num_beams=4,
early_stopping=True
)
summary = tokenizer.decode(outputs[0], skip_special_tokens=True)
st.subheader("Summary:")
st.write(summary)
|