Update app.py

app.py

@@ -69,10 +69,15 @@ def predict(message, history):
         partial_message += chunk
         yield partial_message
         
-gr.Interface(
-    fn=predict, 
-    inputs=[gr.inputs.Textbox(label="Your Message"), gr.inputs.State(label="History")],
-    outputs='text',
+iface = gr.Interface(
+    fn=predict,
+    inputs=gr.Textbox(placeholder="Chat with me!", label="Your Message"),
+    outputs=gr.Text(label="Response"),
     live=True,
-    allow_flagging="never"
-).launch()
+    title="Infectious-Disease-Diagnosis-Chatbot",
+    description="This is the demo for Gradio UI consuming TGI endpoint with LLaMA 7B-Chat model.",
+    examples=[["Are tomatoes vegetables?"]],
+    theme="default"  # You can choose a theme that fits your UI preference
+)
+
+iface.launch()