Update app.py

app.py

@@ -1,11 +1,30 @@
 import gradio as gr
 from transformers import pipeline
 
-classifier = pipeline("text-generation")
-
-gr.Interface(
-    fn=classifier,
-    inputs="text",
-    outputs="text",
-    title="Hugging Face Text Generation"
-).launch(share=True)
+# Load a proper text generation model
+generator = pipeline(
+    "text-generation",
+    model="gpt2",
+    max_length=100,
+    num_return_sequences=1
+)
+
+def generate_text(prompt):
+    result = generator(prompt)
+    return result[0]["generated_text"]
+
+interface = gr.Interface(
+    fn=generate_text,
+    inputs=gr.Textbox(
+        lines=4,
+        placeholder="Enter a prompt...",
+        label="Your Prompt"
+    ),
+    outputs=gr.Textbox(
+        label="Generated Text"
+    ),
+    title="AI Text Generator",
+    description="Enter a prompt and the AI will generate text using GPT-2."
+)
+
+interface.launch()