File size: 796 Bytes
bef7abc 3a33679 bef7abc 026d9d8 bef7abc 026d9d8 bef7abc 026d9d8 bef7abc 026d9d8 bef7abc 026d9d8 bef7abc 026d9d8 |
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 |
import streamlit as st
import transformers
import tensorflow
from transformers import AutoTokenizer
from transformers import TFAutoModelForSeq2SeqLM
model_checkpoint = "Modfiededition/t5-base-fine-tuned-on-jfleg"
tokenizer = AutoTokenizer.from_pretrained(model_checkpoint)
@st.cache
def load_model(model_name):
model = TFAutoModelForSeq2SeqLM.from_pretrained(model_name)
return model
model= 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)
inputs = tokenizer("Grammar: "+sent,return_tensors="tf")
output_ids = model.generate(inputs["input_ids"]).numpy()[0][1:-1]
generated_sequences = tokenizer.decode(output_ids)
st.write(generated_sequences)
|