|
import streamlit as st |
|
import transformers |
|
import tensorflow |
|
|
|
from transformers import pipeline |
|
|
|
|
|
model_checkpoint = "Modfiededition/t5-base-fine-tuned-on-jfleg" |
|
|
|
|
|
@st.cache |
|
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!" |
|
|
|
st.title("Writing Assistant for you π¦") |
|
|
|
sent = st.text_area("Text", default_value, height = 275) |
|
generated_sequences = translator(sent) |
|
|
|
st.write(generated_sequences[-1]) |
|
|