Update app.py

app.py

@@ -1,20 +1,14 @@
 import gradio as gr
 from transformers import pipeline
 
-# Load the Hugging Face model
+# Load a free, public text-generation 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']
+# Define your function
+def respond(prompt):
+    output = pipe(prompt, max_new_tokens=50)[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()
+# Build the interface
+gr.Interface(fn=respond, inputs="text", outputs="text", title="Free AI Essay Bot").launch()
+