Upload 3 files

README.md

@@ -1,14 +1,15 @@
----
-title: Grammar Fixer Phi4 Mini
-emoji: 🐠
-colorFrom: purple
-colorTo: blue
-sdk: gradio
-sdk_version: 5.29.0
-app_file: app.py
-pinned: false
-license: apache-2.0
-short_description: Corrects English grammar using Microsoft's Phi-4 Mini Instru
----
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+# Grammar Fixer (Phi-4 Mini Edition)
+
+A lightweight web app that corrects English grammar using Microsoft's `phi-4-mini-instruct` model.
+
+## Features
+- Instruction-following prompt
+- Fast response
+- Fits Hugging Face Spaces Free Tier
+
+## Model Info
+- [microsoft/phi-4-mini-instruct](https://huggingface.co/microsoft/phi-4-mini-instruct)
+
+## Example Prompt
+Input: `he go to school yesterday`  
+Output: `He went to school yesterday.`

app.py

@@ -0,0 +1,31 @@
+import gradio as gr
+from transformers import AutoTokenizer, AutoModelForCausalLM, TextGenerationPipeline
+import os
+from huggingface_hub import login
+
+# Load token from environment variable
+HUGGINGFACE_TOKEN = os.getenv("HF_TOKEN")
+login(token=HUGGINGFACE_TOKEN)
+
+model_name = "microsoft/phi-4-mini-instruct"
+tokenizer = AutoTokenizer.from_pretrained(model_name, token=HUGGINGFACE_TOKEN)
+model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype="auto", device_map="auto", token=HUGGINGFACE_TOKEN)
+
+pipe = TextGenerationPipeline(model=model, tokenizer=tokenizer)
+
+def correct_grammar(text):
+    prompt = f"Correct the grammar:\n{text}\nCorrected:"
+    result = pipe(prompt, max_new_tokens=64, do_sample=False, temperature=0.3)[0]["generated_text"]
+    if "Corrected:" in result:
+        corrected = result.split("Corrected:")[1].strip()
+    else:
+        corrected = result.strip()
+    return corrected
+
+gr.Interface(
+    fn=correct_grammar,
+    inputs=gr.Textbox(lines=3, placeholder="Enter a grammatically incorrect sentence..."),
+    outputs="text",
+    title="Grammar Fixer",
+    description="Fix grammar using Microsoft's Phi-4 Mini Instruct model."
+).launch()

requirements.txt

@@ -0,0 +1,4 @@
+transformers
+torch
+gradio
+huggingface_hub