File size: 1,025 Bytes
bef7abc 3a33679 bef7abc 712f163 bef7abc 026d9d8 a5f9c84 bef7abc 5c6eae7 4e2fd87 5c6eae7 bef7abc a8d72ba bef7abc 72ef409 bef7abc 45f543a bef7abc d46bc93 3e9d17f 72ef409 3e9d17f 72ef409 3e9d17f 45f543a 3e9d17f |
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 35 36 37 38 39 40 |
import streamlit as st
import transformers
import tensorflow
from transformers import pipeline
model_checkpoint = "Modfiededition/t5-base-fine-tuned-on-jfleg"
@st.cache(allow_output_mutation=True, suppress_st_warning=True)
def load_model():
return pipeline("text2text-generation", model=model_checkpoint)
model = load_model()
#prompts
st.title("Writing Assistant for you 🦄")
st.subheader("Some examples to try: ")
example_1 = st.button("I am write on AI")
example_2 = st.button("This sentence has, bads grammar mistake!")
textbox = st.text_area('Write your text in this box:', '', height=200, max_chars=1000)
button = st.button('Detect grammar mistakes:')
#if example_1:
# output_text = model("I am write on AI")[0]["generated_text]
# st.write(**output_text**)
#if example_2:
# output_text = model("This sentence has, bads grammar mistake!")[0]["generated_text]
# st.write(**output_text**)
#if button:
# output_text = model(textbox)[0]["generated_text]
# st.write(**output_text**)
|