Update app.py

app.py

@@ -2,12 +2,12 @@ import gradio as gr
 from transformers import AutoModelForSeq2SeqLM, AutoTokenizer, GenerationConfig
 
 # Load the model and tokenizer
-model = AutoModelForSeq2SeqLM.from_pretrained("vennify/t5-base-grammar-correction")
-tokenizer = AutoTokenizer.from_pretrained("vennify/t5-base-grammar-correction")
+model = AutoModelForSeq2SeqLM.from_pretrained("pszemraj/flan-t5-large-grammar-synthesis")
+tokenizer = AutoTokenizer.from_pretrained("pszemraj/flan-t5-large-grammar-synthesis")
 
 
 def correct_text(text, genConfig):
-    inputs = tokenizer.encode("grammar: " + text, return_tensors="pt")
+    inputs = tokenizer.encode("" + text, return_tensors="pt")
     outputs = model.generate(inputs, **genConfig.to_dict())
 
     corrected_text = tokenizer.decode(outputs[0], skip_special_tokens=True)