Update app.py

app.py

@@ -3,16 +3,16 @@ from transformers import pipeline
 
 pipeline = pipeline(task="text-generation", model="starcoder2-15b")
 
-def generate_text(input_text):
+def chatbot(input_text):
     generated_text = pipeline(input_text, max_length=50)
-    return input_text, generated_text[0]["generated_text"]
+    return input_text, "Chatbot: " + generated_text[0]["generated_text"]
 
-gradio_app = gr.Interface(
-    generate_text,
-    inputs=gr.Textbox(lines=2, label="Input Text"),
-    outputs=gr.Textbox(label="Generated Text"),
-    title="Text Generation",
+chatbot_app = gr.Interface(
+    chatbot,
+    inputs=gr.Textbox(lines=2, label="You:"),
+    outputs=gr.Textbox(label="Chatbot:", placeholder="Chatbot's response will appear here..."),
+    title="Chatbot UI"
 )
 
 if __name__ == "__main__":
-    gradio_app.launch()
+    chatbot_app.launch()