Update app.py

app.py

@@ -1,17 +1,18 @@
 import gradio as gr
 from transformers import pipeline
 
-pipeline = pipeline(task="text-generation", model="starcoder2-15b")
+generator = pipeline(task="text-generation", model="EleutherAI/gpt-neo-2.7B")
 
 def chatbot(input_text):
-    generated_text = pipeline(input_text, max_length=50)
-    return input_text, "Chatbot: " + generated_text[0]["generated_text"]
+    generated_text = generator(input_text, max_length=50)
+    return generated_text[0]["generated_text"]
 
 chatbot_app = gr.Interface(
-    chatbot,
+    fn=chatbot,
     inputs=gr.Textbox(lines=2, label="You:"),
-    outputs=gr.Textbox(label="Chatbot:", placeholder="Chatbot's response will appear here..."),
-    title="Chatbot UI"
+    outputs="text",
+    title="Chatbot UI",
+    theme="huggingface"
 )
 
 if __name__ == "__main__":