Update app.py

app.py

@@ -1,12 +1,16 @@
 import gradio as gr
-from happytransformer import HappyTextToText, TTSettings
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 
 def check_grammar(sentence):
-  happy_tt = HappyTextToText("T5", "leslyarun/grammatical-error-correction")
-  args = TTSettings(num_beams=5, min_length=1)
-  # Add the prefix "grammar: " before each input 
-  result = happy_tt.generate_text("grammar: " + sentence, args=args)  
-  return result.text
+  tokenizer = AutoTokenizer.from_pretrained("leslyarun/grammatical-error-correction")
+  model = AutoModelForSeq2SeqLM.from_pretrained("leslyarun/grammatical-error-correction")
+  
+  inputs = tokenizer("grammar: " + sentence, truncation=True, return_tensors='pt')
+  
+  output = model.generate(inputs['input_ids'], num_beams=5, max_length=512, early_stopping=True)
+  correction = tokenizer.batch_decode(output, skip_special_tokens=True)
+  return "".join(correction)
+
   
 demo = gr.Interface(check_grammar, inputs=['text'],
                     outputs="text",