Update app.py

app.py

@@ -49,8 +49,8 @@ def generate_response(user_input, chat_history):
     with torch.no_grad():
         outputs = model.generate(**inputs, max_new_tokens=100)
 
-    # Decode response
-    bot_reply = tokenizer.decode(outputs[0], skip_special_tokens=True)
+    # Decode response and remove the instruction part
+    bot_reply = tokenizer.decode(outputs[0], skip_special_tokens=True).replace(instruction, "").strip()
 
     # Update chat history with user and bot interactions
     chat_history.append(("User", user_input))
@@ -63,7 +63,7 @@ def generate_response(user_input, chat_history):
 with gr.Blocks() as demo:
     gr.Markdown("# Chandler-Like Chatbot on GPU")
 
-    chat_history = gr.Chatbot(label="Chat History")
+    chat_history = gr.Chatbot(label="Chat History", elem_id="chatbox")
     user_input = gr.Textbox(
         placeholder="Type your message here...", label="Your Message")
 
@@ -74,4 +74,11 @@ with gr.Blocks() as demo:
     submit_btn.click(generate_response, [user_input, chat_history], [
                      chat_history, user_input])
 
+# Custom CSS to align chat labels on the left
+demo.css = """
+#chatbox .bot, #chatbox .user {
+    text-align: left;
+}
+"""
+
 demo.launch(share=True)  # Enables a public link