Update app.py

app.py

@@ -4,6 +4,7 @@ from huggingface_hub import HfApi
 import spaces
 
 @spaces.GPU
+
 def merge_and_upload(weight_drop_prob, scaling_factor, repo_name, token, base_model, model_to_merge):
     # Define a fixed output path
     output_path = "/home/user/project/output"
@@ -17,8 +18,7 @@ def merge_and_upload(weight_drop_prob, scaling_factor, repo_name, token, base_mo
         "python3", "hf_merge.py",
         "-p", str(weight_drop_prob),
         "-lambda", str(scaling_factor),
-        "-repo_file", "repo.txt",
-        "-output_path", output_path
+        "repo.txt", output_path
     ]
     
     # Run the command and capture the output
@@ -36,8 +36,8 @@ def merge_and_upload(weight_drop_prob, scaling_factor, repo_name, token, base_mo
         
         # Upload the file
         api.upload_file(
-            path_or_fileobj=output_path + "/merged_model.safetensor",
-            path_in_repo="merged_model.safetensor",
+            path_or_fileobj=output_path,
+            path_in_repo="merged_model.safetensors",
             repo_id=repo_name,
             token=token
         )