Update app.py

app.py

@@ -70,8 +70,7 @@ def merge_models() -> None:
     """
     Use mergekit to create a merge.
     """
-    # command = ["mergekit-yaml", "config.yaml", "/data/merge", "--copy-tokenizer", "--transformers_cache", "/data"]
-    command = ["mergekit-yaml", "config.yaml", "/data/merge", "--copy-tokenizer"]
+    command = ["mergekit-yaml", "config.yaml", "/data/merge", "--copy-tokenizer", "--transformers_cache", "/data"]
 
     with open("output.log", "a") as log_file: 
         try:
@@ -79,7 +78,7 @@ def merge_models() -> None:
                                     stderr=log_file, text=True)
             print(f"mergekit: {result.stdout}")
         except subprocess.CalledProcessError as e:
-            print(f"Error: mergekit: {e.stderr}")
+            print(f"Error: mergekit {command}: {e.stderr}")
 
 
 def make_df(file_path: str, n_rows: int) -> pd.DataFrame: