Upload 2 files

app.py

@@ -0,0 +1,81 @@
+import gradio as gr
+from transformers import pipeline
+import os
+
+# --- Model Loading ---
+# We use the 'text2text-generation' pipeline. 
+# device=-1 forces CPU usage, which is required for the free tier unless you have a GPU upgrade.
+print("Initializing model...")
+
+try:
+    # Attempt to load the specific grammar correction model
+    corrector = pipeline(
+        "text2text-generation",
+        model="HamadML/grammer_correction",
+        device=-1
+    )
+    print("Model loaded successfully.")
+except Exception as e:
+    print(f"Error loading model: {e}")
+    corrector = None
+
+# --- Core Logic ---
+def correct_grammar(text):
+    """
+    Takes input text and returns the corrected version using the loaded model.
+    """
+    if not text or not text.strip():
+        return ""
+    
+    if corrector is None:
+        return "Error: The model failed to load. Please check the 'Logs' tab in your Space."
+
+    try:
+        # Generate correction
+        # max_length=128 ensures we don't run out of memory on long sentences
+        results = corrector(text, max_length=128)
+        
+        # The pipeline returns a list of dicts: [{'generated_text': '...'}]
+        corrected_text = results[0]['generated_text']
+        return corrected_text
+    except Exception as e:
+        return f"Error during processing: {str(e)}"
+
+# --- User Interface ---
+# Using Gradio Blocks for a custom layout
+with gr.Blocks(title="Grammar Corrector", theme=gr.themes.Soft()) as demo:
+    gr.Markdown(
+        """
+        # 🇬🇧 Grammar Correction AI
+        Type a sentence below to fix grammar, spelling, and punctuation errors.
+        """
+    )
+    
+    with gr.Row():
+        with gr.Column():
+            input_box = gr.Textbox(
+                label="Input Text", 
+                placeholder="Example: I has went to the store yesterday...",
+                lines=4
+            )
+            submit_btn = gr.Button("Fix Grammar", variant="primary")
+        
+        with gr.Column():
+            output_box = gr.Textbox(
+                label="Corrected Output", 
+                lines=4,
+                interactive=False,
+                show_copy_button=True
+            )
+
+    # Event listeners
+    submit_btn.click(fn=correct_grammar, inputs=input_box, outputs=output_box)
+    input_box.submit(fn=correct_grammar, inputs=input_box, outputs=output_box)
+
+    gr.Markdown(
+        "Powered by [HamadML/grammer_correction](https://huggingface.co/HamadML/grammer_correction)"
+    )
+
+# Launch the app
+if __name__ == "__main__":
+    demo.launch()

requirements.txt

@@ -0,0 +1,5 @@
+transformers>=4.30.0
+torch
+gradio
+sentencepiece
+protobuf