Update app.py

app.py

@@ -15,6 +15,7 @@ name2repo = [("Training", "bigcode-project/Megatron-LM"),
              # ("Inference", "bigcode-project/bigcode-inference-benchmark"),
             ]
 
+name2num_milestones = dict()
 g = Github(os.environ.get('github'))
 df = list()
 all_status = list()
@@ -36,6 +37,7 @@ for name, repo_name in name2repo:
                        Finish=milestone.due_on.strftime('%Y-%m-%d'), 
                        Resource=desc['status'], 
                        Description=desc['leader']))
+    name2num_milestones[name] = len(repo.get_milestones())
 
 colors = {'not started': 'rgb(217, 217, 217)',
           'in progress': 'rgb(147, 196, 125)',
@@ -88,7 +90,11 @@ fig.add_annotation({
             "showarrow": False,
         })
 
+
 fig.add_hline(y=-0.5, line_width=1, line_color="grey")
-fig.add_hline(y=2.5, line_width=1, line_color="grey")
+height = -0.5
+for name, _ in name2repo[::-1]:
+    height += name2num_milestones[name]
+    fig.add_hline(y=height, line_width=1, line_color="grey")
 
 st.plotly_chart(fig, use_container_width=True)