Create app.py

app.py

@@ -0,0 +1,22 @@
+import gradio as gr
+from transformers import pipeline
+
+# Load GPT-2 chat model
+chatbot = pipeline("text-generation", model="gpt2")
+
+# Function to handle the chatbot interaction
+def chat_with_gpt2(messages):
+    prompt = messages[-1][1]  # Get the latest message from the user
+    response = chatbot(prompt, max_length=100, num_return_sequences=1)
+    return messages + [(response[0]['generated_text'], "GPT-2")]
+
+# Create a Gradio Chatbot interface
+iface = gr.Interface(
+    fn=chat_with_gpt2,
+    inputs=gr.Chatbot(),
+    outputs=gr.Chatbot(),
+    title="Chat with GPT-2",
+    description="A simple chat app using GPT-2"
+)
+
+iface.launch()