Update app.py

app.py

@@ -400,7 +400,8 @@ def merge_loop():
     df = make_df("open-llm-leaderboard.csv", N_ROWS)
 
     # Sample two models
-    print(f"Total size: {human_readable_size(get_size('/'))}")
+    dir_path = "~/.cache/huggingface/hub"
+    print(f"Total size: {human_readable_size(get_size(dir_path))}")
     sample = df.sample(n=2)
     models = [sample.iloc[i] for i in range(2)]
 
@@ -416,30 +417,30 @@ def merge_loop():
     # Merge configs
     yaml_config = create_config(models)
     print(f"YAML config:{yaml_config}")
-    print(f"Total size: {human_readable_size(get_size('/'))}")
+    print(f"Total size: {human_readable_size(get_size(dir_path))}")
 
     # Merge models
     merge_models()
     print("Model merged!")
-    print(f"Total size: {human_readable_size(get_size('/'))}")
+    print(f"Total size: {human_readable_size(get_size(dir_path))}")
 
     # Create model card
     print("Create model card")
     create_model_card(yaml_config, model_name, USERNAME, license)
-    print(f"Total size: {human_readable_size(get_size('/'))}")
+    print(f"Total size: {human_readable_size(get_size(dir_path))}")
 
     # Upload model
     print("Upload model")
     upload_model(api, USERNAME, model_name)
-    print(f"Total size: {human_readable_size(get_size('/'))}")
+    print(f"Total size: {human_readable_size(get_size(dir_path))}")
 
     # Clear data
     clear_data()
-    print(f"Total size: {human_readable_size(get_size('/'))}")
+    print(f"Total size: {human_readable_size(get_size(dir_path))}")
     
     # Evaluate model on Runpod
     create_pod(model_name, USERNAME)
-    print(f"Total size: {human_readable_size(get_size('/'))}")
+    print(f"Total size: {human_readable_size(get_size(dir_path))}")
 
     
 # Install scrape-open-llm-leaderboard and mergekit