Update app.py

app.py

@@ -37,4 +37,18 @@ def predict(input, history=[]):
     history = response[Completion]
     
     # convert the tokens to text, and then split the responses into lines
-    response = tokenizer.decode(history[0]).split("
+    response = tokenizer.decode(history[0]).split
+
+css = """
+.chatbox {display:flex;flex-direction:row}
+.msg {padding:4px;margin-bottom:4px;border-radius:4px;width:80%}
+.msg.user {background-color:cornflowerblue;color:white}
+.msg.bot {background-color:lightgray;align-self:self-end}
+.footer {display:none !important}
+"""
+
+gr.Interface(fn=predict,
+             theme="default",
+             inputs=[gr.inputs.Textbox(placeholder="I'm AI-DHD - ask me anything!"), "state"],
+             outputs=["html", "state"],
+             css=css).launch()