Update app.py

app.py

@@ -13,12 +13,11 @@ def chat(message, token_response):
     response = tokenizer.decode(token_response[:, token_message.shape[-1]:][0], skip_special_tokens=True)
     return response, token_response
 
-input = gr.inputs.Textbox(lines=2, label='User:')
-output = gr.outputs.Textbox(label='Elon:')
-
-gr.Interface(fn=chat,
-    title="Chat with Elon",
-    inputs=[input, "state"],
-    outputs=[output, "state"],
-    allow_screenshot=False,
-    allow_flagging='never').launch()
+gr.Interface(fn=predict,
+             title="Chat with Elon",
+             theme="default",
+             css=".footer {display:none !important}",
+             allow_screenshot=False,
+             allow_flagging='never'
+             inputs=["User", "state"],
+             outputs=["Elon", "state"]).launch()