Update app.py

app.py

@@ -9,27 +9,24 @@ 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)
+@st.cache(allow_output_mutation=True, suppress_st_warning=True)
+def load_model():
+    return pipeline("text2text- generation", model=model_checkpoint)
+model = load_model()
 
 def infer(input_ids):
     output_sequences = model.generate(inputs["input_ids"]).numpy()[0][1:-1]
     return output_sequences
     
-default_value = "Write your text here!"
 #prompts
 st.title("Writing Assistant for you 🦄")
 
-sent = st.text_area("Text", default_value, height = 275)
+textbox = st.text_area('Write your text:', '', height=200, max_chars=1000)
 
-inputs = tokenizer("Grammar: "+sent,return_tensors="tf")
+#inputs = tokenizer("Grammar: "+sent,return_tensors="tf")
 
-output_sequences = infer(inputs)
+#output_sequences = infer(inputs)
 
-generated_sequences = tokenizer.decode(output_ids)
+#generated_sequences = tokenizer.decode(output_ids)
 
-st.write(generated_sequences)
+#st.write(generated_sequences)