import streamlit as st
from t5_podcast_summariser import Summariser


st.text_input("Transcript:", key="transcript")
st.write("Input: \n" + st.session_state.transcript + "\n")

st.write("Loading model...")
summariser = Summariser()
st.write("Model loading complete...")

st.write("Starting Summarisation Inference...")
summary = summariser.summarise(st.session_state.transcript)
st.write("Summary:")
st.write(summary)
st.write("---------------------------------------------")