Update app.py

app.py

@@ -105,12 +105,13 @@ def upload_stage1_result(stage1_param_path: str, stage1_result_path: str,
     if not UPLOAD_REPO_ID:
         return
     try:
+        random_folder = random.randint(0,1000)
         hf_api.upload_file(path_or_fileobj=stage1_param_path,
-                           path_in_repo=f'stage1_params/{save_name}.json',
+                           path_in_repo=f'stage1_params/{random_folder}/{save_name}.json',
                            repo_id=UPLOAD_REPO_ID,
                            repo_type='dataset')
         hf_api.upload_file(path_or_fileobj=stage1_result_path,
-                           path_in_repo=f'stage1_results/{save_name}.pth',
+                           path_in_repo=f'stage1_results/{random_folder}/{save_name}.pth',
                            repo_id=UPLOAD_REPO_ID,
                            repo_type='dataset')
     except Exception as e:
@@ -141,13 +142,14 @@ def upload_stage2_info(stage1_param_file_hash_name: str,
     save_name = f'{stage1_param_file_hash_name}_{stage2_param_file_hash_name}'
 
     try:
+        random_folder = random.randint(0,1000)
         hf_api.upload_file(path_or_fileobj=param_file.name,
-                           path_in_repo=f'stage2_params/{save_name}.json',
+                           path_in_repo=f'stage2_params/{random_folder}/{save_name}.json',
                            repo_id=UPLOAD_REPO_ID,
                            repo_type='dataset')
         if UPLOAD_RESULT_IMAGE:
             hf_api.upload_file(path_or_fileobj=stage2_output_path,
-                               path_in_repo=f'stage2_results/{save_name}.png',
+                               path_in_repo=f'stage2_results/{random_folder}/{save_name}.png',
                                repo_id=UPLOAD_REPO_ID,
                                repo_type='dataset')
     except Exception as e:
@@ -186,8 +188,9 @@ def upload_stage2_3_info(stage1_param_file_hash_name: str,
     save_name = f'{stage1_param_file_hash_name}_{stage2_3_param_file_hash_name}'
 
     try:
+        random_folder = random.randint(0,1000)
         hf_api.upload_file(path_or_fileobj=param_file.name,
-                           path_in_repo=f'stage2_3_params/{save_name}.json',
+                           path_in_repo=f'stage2_3_params/{random_folder}/{save_name}.json',
                            repo_id=UPLOAD_REPO_ID,
                            repo_type='dataset')
         if UPLOAD_RESULT_IMAGE: