Update app.py

app.py

@@ -1,24 +1,42 @@
 from transformers import pipeline, set_seed
 import gradio as gr
 
-generator = pipeline('text-generation', model='Keyurjotaniya007/openai-gpt2-wikitext-textgen')
+generator = pipeline(
+    "text-generation",
+    model="Keyurjotaniya007/openai-gpt2-wikitext-textgen"
+)
+
 set_seed(42)
 
 def generate_text(prompt):
+    if not prompt.strip():
+        return "Please enter a prompt to generate text."
     response = generator(
         prompt,
-        max_length=100,
+        max_length=150,
         num_return_sequences=1,
+        temperature=0.8,
+        top_p=0.95,
+        do_sample=True,
         truncation=True
     )
-    return response[0]['generated_text']
+    return response[0]["generated_text"]
 
+# Define the Gradio interface
 interface = gr.Interface(
     fn=generate_text,
-    inputs=gr.Textbox(lines=5, placeholder="write something to generate..."),
-    outputs="text",
-    title="OpenAI GPT-2🦄",
-    description="Run text generation with the model `Keyurjotaniya007/openai-gpt2-wikitext-textgen`, a GPT-2 variant fine-tuned for coherent long-form text."
+    inputs=gr.Textbox(lines=4, placeholder="Type your prompt here..."),
+    outputs=gr.Textbox(label="Generated Text"),
+    title="🦄 GPT-2 WikiText Text Generator",
+    description=(
+        "Generate coherent, fluent English text using **GPT-2 fine-tuned on the WikiText dataset**. "
+        "Model: `Keyurjotaniya007/openai-gpt2-wikitext-textgen`"
+    ),
+    examples=[
+        ["Artificial intelligence is transforming the world by"],
+        ["Once upon a time, in a small village,"],
+        ["The future of natural language processing lies in"],
+    ],
 )
 
 if __name__ == "__main__":