Update app.py

app.py

@@ -9,6 +9,13 @@ from collections import defaultdict
 from networkx.drawing.nx_pydot import graphviz_layout
 from io import BytesIO
 
+TITLE = """
+<div align="center">
+    <p style="font-size: 36px;">🌳 Merge Family Tree</p>
+</div><br/>
+<p>Automatically calculate the family tree of a given model. It also displays the type of license each model uses (permissive, noncommercial, or unknown).</p>
+"""
+
 
 def get_model_names_from_yaml(url):
     """Get a list of parent model names from the yaml file."""
@@ -165,7 +172,7 @@ def create_family_tree(start_model):
     return img_buffer.getvalue()
     
 with gr.Blocks() as demo:
-    gr.Markdown("# 🌳 Merge Family Tree")
+    gr.Markdown(TITLE)
     model_id = gr.Textbox(label="Model ID", value="mlabonne/NeuralBeagle-7B")
     btn = gr.Button("Create tree")
     out = gr.Image()