diff --git a/demo/chatbot/run.py b/demo/chatbot/run.py
index d8eedb630e..a465721031 100644
--- a/demo/chatbot/run.py
+++ b/demo/chatbot/run.py
@@ -14,24 +14,13 @@ def chat(message, history):
     else:
         response = "I don't know"
     history.append((message, response))
-    html = "<div class='chatbot'>"
-    for user_msg, resp_msg in history:
-        html += f"<div class='user_msg'>{user_msg}</div>"
-        html += f"<div class='resp_msg'>{resp_msg}</div>"
-    html += "</div>"
-    return html, history
+    return history, history
 
 
 iface = gr.Interface(
     chat,
     ["text", "state"],
-    ["html", "state"],
-    css="""
-    .chatbox {display:flex;flex-direction:column}
-    .user_msg, .resp_msg {padding:4px;margin-bottom:4px;border-radius:4px;width:80%}
-    .user_msg {background-color:cornflowerblue;color:white;align-self:start}
-    .resp_msg {background-color:lightgray;align-self:self-end}
-""",
+    ["chatbot", "state"],
     allow_screenshot=False,
     allow_flagging="never",
 )
diff --git a/demo/xray_blocks/run.py b/demo/xray_blocks/run.py
index 79533c2389..c9d9db858d 100644
--- a/demo/xray_blocks/run.py
+++ b/demo/xray_blocks/run.py
@@ -3,7 +3,7 @@ import gradio as gr
 import random
 
 xray_model = lambda diseases, img: {disease: random.random() for disease in diseases}
-ct_model = lambda diseases, img: {disease: random.random() for disease in diseases}
+ct_model = lambda diseases, img: {disease: 0.1 for disease in diseases}
 
 xray_blocks = gr.Blocks()