Update app.py

app.py

@@ -402,7 +402,6 @@ def merge_loop():
     # Sample two models
     dir_path = "/home/"
     print(f"pwd: {os.getcwd()}")
-    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)]
 
@@ -418,30 +417,32 @@ 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(dir_path))}")
+    # 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(dir_path))}")
+    # 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(dir_path))}")
+    # 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(dir_path))}")
+    # print(f"Total size: {human_readable_size(get_size(dir_path))}")
 
     # Clear data
+    print("Clear data")
     clear_data()
-    print(f"Total size: {human_readable_size(get_size(dir_path))}")
+    # print(f"Total size: {human_readable_size(get_size(dir_path))}")
     
     # Evaluate model on Runpod
+    print("Start evaluation")
     create_pod(model_name, USERNAME)
-    print(f"Total size: {human_readable_size(get_size(dir_path))}")
+    # print(f"Total size: {human_readable_size(get_size(dir_path))}")
 
     
 # Install scrape-open-llm-leaderboard and mergekit