app.py

import gradio as gr
from transformers import pipeline

# Load the Hugging Face model
pipe = pipeline("text-generation", model="mrm8488/t5-base-finetuned-common_gen")

# Define what the app does
def rewrite(text):
    prompt = f"Fix grammar and improve clarity for this text:\n\n{text}"
    output = pipe(prompt, max_new_tokens=300, do_sample=True)[0]['generated_text']
    return output

# Create the interface
gr.Interface(
    fn=rewrite,
    inputs=gr.Textbox(lines=10, placeholder="Paste your essay here..."),
    outputs=gr.Textbox(lines=10, label="Improved Essay"),
    title="AI Essay Rewriter",
    description="Paste an essay or paragraph and get an improved version using Mistral-7B."
).launch()