Update app.py

app.py

@@ -70,7 +70,7 @@ def merge_models() -> None:
     """
     Use mergekit to create a merge.
     """
-    command = ["mergekit-yaml", "config.yaml", "merge", "--copy-tokenizer"]
+    command = ["mergekit-yaml", "config.yaml", "/data/merge", "--copy-tokenizer"]
 
     with open("output.log", "a") as log_file: 
         try:
@@ -261,7 +261,7 @@ print(outputs[0]["generated_text"])
 
     # Save the model card
     card = ModelCard(content)
-    card.save('merge/README.md')
+    card.save('data/merge/README.md')
 
 
 def upload_model(api: HfApi, username: str, model_name: str) -> None:
@@ -275,7 +275,7 @@ def upload_model(api: HfApi, username: str, model_name: str) -> None:
     )
     api.upload_folder(
         repo_id=f"{username}/{model_name}",
-        folder_path="merge",
+        folder_path="/data/merge",
     )
 
 
@@ -358,7 +358,7 @@ def clear_data():
     """
     Clear data so the Space doesn't crash...
     """
-    dir_path = "merge"
+    dir_path = "/data/merge"
     if os.path.exists(dir_path) and os.path.isdir(dir_path):
         shutil.rmtree(dir_path)
         print(f"The directory '{dir_path}' has been removed successfully.")
@@ -400,7 +400,7 @@ def merge_loop():
     df = make_df("open-llm-leaderboard.csv", N_ROWS)
 
     # Sample two models
-    dir_path = "/home/"
+    dir_path = "/data"
     print(f"pwd: {os.getcwd()}")
     sample = df.sample(n=2)
     models = [sample.iloc[i] for i in range(2)]