feat: update output code

web_server.py

@@ -28,7 +28,7 @@ IN_HUGGINGFACE_SPACE = os.getenv('IN_HUGGINGFACE_SPACE', 'false') == 'true'
 STOP_SPACE_TIMEOUT = int(os.getenv('STOP_SPACE_TIMEOUT', '7200'))
 HF_TOKEN = os.getenv('HF_TOKEN', None)
 SPACE_PARAMS = json.loads(os.getenv('PARAMS', '{}'))
-OUTPUT_DIR = "/app/app_datas/output"
+OUTPUT_DIR = "/app/app_datas/env_output"
 
 print("IN_HUGGINGFACE_SPACE:", IN_HUGGINGFACE_SPACE)
 print("STOP_SPACE_TIMEOUT:", STOP_SPACE_TIMEOUT)
@@ -109,7 +109,7 @@ def auto_stop():
     if GlobalState.done:
         api.upload_folder(
             repo_id=SPACE_PARAMS["competition_id"],
-            folder_path=OUTPUT_DIR,
+            folder_path=os.path.join(OUTPUT_DIR, "hugsim_env"),
             repo_type="dataset",
             path_in_repo=f"eval_results/{SPACE_PARAMS['submission_id']}",
         )
@@ -350,11 +350,13 @@ if __name__ == "__main__":
     model_config = OmegaConf.load(os.path.join(model_path, 'cfg.yaml'))
     model_config.update({"model_path": "/app/app_datas/PAMI2024/release/ss/scenes/nuscenes/scene-0383"})
     cfg.update(model_config)
+    cfg.base.output_dir = OUTPUT_DIR
     
-    os.makedirs(OUTPUT_DIR, exist_ok=True)
+    output = os.path.join(OUTPUT_DIR, "hugsim_env")
+    os.makedirs(output, exist_ok=True)
     
-    print("Output directory:", OUTPUT_DIR)
-    env_handler = EnvHandler(cfg, OUTPUT_DIR)
+    print("Output directory:", output)
+    env_handler = EnvHandler(cfg, output)
     print("Environment handler initialized.")
     web_server = WebServer(env_handler, auth_token=os.getenv('HUGSIM_AUTH_TOKEN'))
     print("Web server initialized.")