import streamlit as st
from happytransformer import HappyTextToText, TTSettings

happy_tt = HappyTextToText("T5", "team-writing-assistant/t5-base-c4jfleg")
args = TTSettings(num_beams=5, min_length=1)

input = st.text_area('Enter some text')

if input:
  input = "grammar: " + input
  result = happy_tt.generate_text(input, args=args)
  st.text(result.text)