Update app.py

app.py

@@ -4,11 +4,10 @@ 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.
+# device=-1 forces CPU usage, which is required for the free tier.
 print("Initializing model...")
 
 try:
-    # Attempt to load the specific grammar correction model
     corrector = pipeline(
         "text2text-generation",
         model="HamadML/grammer_correction",
@@ -21,33 +20,26 @@ except Exception as e:
 
 # --- 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."
+        return "Error: The model failed to load."
 
     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:
+    # UPDATED: Removed Emoji and "GB"
     gr.Markdown(
         """
-        # 🇬🇧 Grammar Correction AI
-        Type a sentence below to fix grammar, spelling, and punctuation errors.
+        # Grammar Correction AI
+        Type a sentence below to check your text.
         """
     )
     
@@ -55,27 +47,23 @@ with gr.Blocks(title="Grammar Corrector", theme=gr.themes.Soft()) as demo:
         with gr.Column():
             input_box = gr.Textbox(
                 label="Input Text", 
-                placeholder="Example: I has went to the store yesterday...",
+                placeholder="Example: I has went to the store...",
                 lines=4
             )
-            submit_btn = gr.Button("Fix Grammar", variant="primary")
+            # UPDATED: Button text changed
+            submit_btn = gr.Button("Check My Text", variant="primary")
         
         with gr.Column():
-            # Removed show_copy_button=True to fix TypeError
             output_box = gr.Textbox(
                 label="Corrected Output", 
                 lines=4,
                 interactive=False
             )
 
-    # 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)"
-    )
+    gr.Markdown("Powered by HamadML/grammer_correction")
 
-# Launch the app
 if __name__ == "__main__":
     demo.launch()