Spaces:
Runtime error
Runtime error
File size: 1,746 Bytes
b69b754 |
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 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 |
import time
import errant
import streamlit as st
from happytransformer import HappyTextToText, TTSettings
from highlighter import show_edits, show_highlights
checkpoints = [
"aseifert/t5-base-jfleg-wi",
"aseifert/byt5-base-jfleg-wi",
"prithivida/grammar_error_correcter_v2",
"Modfiededition/t5-base-fine-tuned-on-jfleg",
]
@st.cache(suppress_st_warning=True, allow_output_mutation=True)
def get_model(model_name):
return HappyTextToText("T5", model_name)
@st.cache(suppress_st_warning=True, allow_output_mutation=True)
def get_annotator(lang: str):
return errant.load(lang)
st.title("Check & Improve English Grammar")
st.markdown("This writing assistant detects π and corrects βοΈ grammatical mistakes for you!")
checkpoint = st.selectbox("Choose model", checkpoints)
happy_tt = get_model(checkpoint)
annotator = get_annotator("en")
args = TTSettings(num_beams=5, min_length=1, max_length=1024)
input_text = st.text_area(
label="Original text",
value="A dog is bigger then mouse.",
placeholder="Enter your text here",
)
button = st.button("βοΈ Check")
def output(input_text):
with st.spinner("Checking for errors π"):
prefixed_input_text = "Grammar: " + input_text
result = happy_tt.generate_text(prefixed_input_text, args=args).text
try:
st.success(result)
show_highlights(annotator, input_text, result)
# st.table(show_edits(annotator, input_text, result))
except Exception as e:
st.error("Some error occured!" + str(e))
st.stop()
start = time.time()
output(input_text)
st.write("---")
st.text(f"Built by Team Writing Assistant β€οΈ β prediction took {time.time() - start:.2f}s")
|