Better presentation

main.py

@@ -128,26 +128,40 @@ with gr.Blocks(js=JS) as demo:
         show_label=True,
         container=True,
         elem_classes="auto-scroll",
-        max_height="90vh",
         render=False,
+        height="80vh",
     )
-    with gr.Row(equal_height=True, height="90vh"):
+    with gr.Row(equal_height=True):
         with gr.Column(scale=3):
             gr.ChatInterface(
                 chat,
                 type="messages",
                 chatbot=chat_bot,
-                title=str(model_name),
+                title="Simple conversational AI with reasoning",
                 description=(
-                    f"*{model_name}* is a large language model "
+                    f"We're using the **{model_name}**. It is a large language model "
                     "trained on a mixture of instruction and "
-                    "conversational data."
+                    "conversational data. It has the capability to reason about the "
+                    "prompt (the user question). "
+                    "When you ask a question, you can see its thoughts "
+                    "on the left block."
                 ),
                 additional_outputs=[reasoning],
             )
 
         with gr.Column():
             reasoning.render()
+            gr.Markdown(
+                "\n\n"
+                "Note that we are deliberately using a small model with 1.5B parameters. "
+                "In spite of this, the model's reasoning capability offers highly relevant "
+                "answer generation. This interface can run on a personal computer if you have 6GB of GPU memory."
+                "\n\n"
+                "See [The model page](https://huggingface.co/deepseek-ai/DeepSeek-R1-Distill-Qwen-1.5B) "
+                "for more information.",
+                container=True,
+                max_height="20vh",
+            )
 
 
 if __name__ == "__main__":