import streamlit as st
import transformers
import tensorflow

from transformers import pipeline

# Replace this with your own checkpoint
model_checkpoint = "Modfiededition/t5-base-fine-tuned-on-jfleg"


def load_model(model_name):
    translator= pipeline("text2text-generation", model=model_checkpoint)
    return translator
    
translator= load_model(model_checkpoint)

default_value = "Write your text here!"
#prompts
st.title("Writing Assistant for you 🦄")

sent = st.text_area("Text", default_value, height = 275)
generated_sequences = translator(sent)

st.write(generated_sequences[-1])