Update app.py

app.py

@@ -118,7 +118,7 @@ def calculate_width_at_depth(G, root):
     return depth_count
 
 
-def create_family_tree(start_model, progress=gr.Progress()):
+def create_family_tree(start_model):
     genealogy = defaultdict(list)
     get_model_names(start_model, genealogy)  # Assuming this populates the genealogy
 
@@ -126,14 +126,8 @@ def create_family_tree(start_model, progress=gr.Progress()):
     G = nx.DiGraph()
 
     # Add nodes and edges to the graph
-    progress.total = len(genealogy.items())
-    items_done = 0
     for parent, children in genealogy.items():
         for child in children:
-            items_done += 1
-            if items_done >= len(genealogy.items()):
-                progress.progress += 1
-                progress.total = items_done
             G.add_edge(parent, child)
 
     # Get max depth