grammar model changed

app/services/grammar.py

@@ -1,8 +1,8 @@
-import language_tool_python
+from transformers import pipeline
 
-tool = language_tool_python.LanguageToolPublicAPI('en-US')
+grammar_corrector = pipeline("text2text-generation", model="pszemraj/flan-t5-large-grammar-synthesis")
 
 def correct_grammar(text: str) -> str:
-    matches = tool.check(text)
-    corrected_text = language_tool_python.utils.correct(text, matches)
-    return corrected_text
+    prompt = "grammar: " + text
+    result = grammar_corrector(prompt, max_length=min(len(prompt.split()) + 50, 128), do_sample=False)
+    return result[0]["generated_text"]  

requirements.txt

@@ -4,5 +4,4 @@ transformers
 torch
 pydantic
 protobuf
-huggingface-hub
-language_tool_python
+huggingface-hub