[refactor] set vis_output folder from the env VIS_OUTPUT with default under ROOT project

lisa_on_cuda/utils/app_helpers.py

@@ -27,7 +27,7 @@ def parse_args(args_to_parse):
     logging.info(f"ROOT_PROJECT:{utils.PROJECT_ROOT_FOLDER}.")
     parser = argparse.ArgumentParser(description="LISA chat")
     parser.add_argument("--version", default="xinlai/LISA-13B-llama2-v1-explanatory")
-    parser.add_argument("--vis_save_path", default=str(Path(utils.PROJECT_ROOT_FOLDER) / "vis_output"), type=str)
+    parser.add_argument("--vis_save_path", default=str(utils.VIS_OUTPUT), type=str)
     parser.add_argument(
         "--precision",
         default="fp16",

lisa_on_cuda/utils/utils.py

@@ -46,6 +46,7 @@ ANSWER_LIST = [
 ]
 ROOT = Path(__file__).parent.parent.parent
 PROJECT_ROOT_FOLDER = os.getenv("PROJECT_ROOT_FOLDER", ROOT)
+VIS_OUTPUT = os.getenv("VIS_OUTPUT", ROOT / "vis_output")
 
 
 class Summary(Enum):