Update app.py

app.py

@@ -96,11 +96,8 @@ def create_stats_html():
     
     # Create derivative type distribution chart
     if len(derivative_df) > 0:
-        # Create distribution by model type
-        type_dist = derivative_df.groupby('model_type').agg({
-            'model_id': 'count',
-            'downloads_30d': 'sum'
-        }).reset_index()
+        # Group by model_type and sum downloads
+        type_dist = derivative_df.groupby('model_type')['downloads_30d'].sum().reset_index()
         
         # Format model types to be more readable
         type_dist['model_type'] = type_dist['model_type'].str.capitalize()
@@ -110,7 +107,7 @@ def create_stats_html():
             type_dist, 
             x='model_type',    
             y='downloads_30d', 
-            title='Downloads by Model Type',
+            title='Total Downloads by Model Type',
             labels={'downloads_30d': 'Downloads', 'model_type': 'Model Type'},
             text=type_dist['downloads_30d'].apply(format_number)
         )