Update app.py

app.py

@@ -1,23 +1,21 @@
+from transformers import pipeline
 import gradio as gr
-from transformers import pipeline, set_seed
 
+# Load GPT-2 (or use sshleifer/tiny-gpt2 for low RAM)
 generator = pipeline("text-generation", model="sshleifer/tiny-gpt2")
-set_seed(42)
 
-def chat_fn(message):
-    result = generator(message, max_length=60, num_return_sequences=1)
-    reply = result[0]["generated_text"].replace(message, "").strip()
+def chat_fn(message, history=[]):
+    prompt = message.strip()
+    result = generator(prompt, max_length=60, num_return_sequences=1)
+    reply = result[0]['generated_text'].replace(prompt, "").strip()
     return reply
 
-# Use Blocks for better layout control
-with gr.Blocks(css="body {margin-top: 40px;}") as demo:
-    gr.Markdown("# 💬 GPT-2 Chatbot")
-    gr.Markdown("Talk to a small GPT-2 model powered by Transformers and Gradio.")
-    with gr.Row():
-        with gr.Column(scale=4):
-            user_input = gr.Textbox(placeholder="Type your message...", label="You")
-            output = gr.Textbox(label="Bot")
-            btn = gr.Button("Send")
-            btn.click(fn=chat_fn, inputs=user_input, outputs=output)
+demo = gr.ChatInterface(
+    fn=chat_fn,
+    title="💬 GPT-2 Chatbot",
+    description="Powered by sshleifer/tiny-gpt2 for light, fast, local inference"
+)
+
+if __name__ == "__main__":
+    demo.launch(share=True)
 
-demo.launch()