Update app.py

app.py

@@ -1,7 +1,25 @@
 import gradio as gr
+from transformers import pipeline
 
-def greet(name):
-    return "Hello " + name + "!!"
+# Load the GPT-2 model
+gpt2 = pipeline("text-generation", model="gpt2")
 
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
-iface.launch()
+# Define the function that will generate the response
+def chat_with_gpt2(user_input):
+    # Use the GPT-2 model to generate a response
+    responses = gpt2(user_input, max_length=100, num_return_sequences=1)
+    # Return the first (and only) response
+    return responses[0]['generated_text']
+
+# Create the Gradio interface
+iface = gr.Interface(
+    fn=chat_with_gpt2, 
+    inputs=gr.inputs.Textbox(lines=2, placeholder="Type your message here..."), 
+    outputs="text",
+    title="Chat with GPT-2",
+    description="This is a simple chatbot powered by GPT-2. Type your message and get a response."
+)
+
+# Launch the app
+if __name__ == "__main__":
+    iface.launch()