Update app.py

app.py

@@ -1,23 +1,25 @@
 import gradio as gr
-import language_tool_python
+from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
+import torch
 
-# Load English grammar checking tool
-tool = language_tool_python.LanguageTool('en-US')
+# Load grammar correction model
+model_name = "pszemraj/grammar-synthesis-small"
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
 
-# Function to check grammar
-def grammar_check(text):
-    matches = tool.check(text)
-    corrected = language_tool_python.utils.correct(text, matches)
-    return corrected, len(matches)
+# Grammar correction function
+def correct_grammar(text):
+    input_text = "gec: " + text
+    inputs = tokenizer.encode(input_text, return_tensors="pt", truncation=True)
+    outputs = model.generate(inputs, max_length=512, num_beams=5, early_stopping=True)
+    corrected_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
+    return corrected_text
 
-# Gradio UI
+# Gradio Interface
 gr.Interface(
-    fn=grammar_check,
-    inputs=gr.Textbox(lines=7, placeholder="Type or paste your text here...", label="Input Text"),
-    outputs=[
-        gr.Textbox(label="Corrected Text"),
-        gr.Number(label="Number of Issues")
-    ],
-    title="Grammar Checker",
-    description="A simple tool to check and correct grammar using LanguageTool."
+    fn=correct_grammar,
+    inputs=gr.Textbox(lines=7, placeholder="Enter your text here...", label="Input Text"),
+    outputs=gr.Textbox(label="Corrected Text"),
+    title="Grammar Checker (No Java)",
+    description="Uses a Hugging Face transformer model to fix grammar mistakes in English."
 ).launch()