[feat] add ROOT_PROJECT ENV variable

lisa_on_cuda/__version__.py

@@ -0,0 +1 @@
+__version__ = "0.2.0"

lisa_on_cuda/utils/app_helpers.py

@@ -2,6 +2,7 @@ import argparse
 import logging
 import os
 import re
+from pathlib import Path
 from typing import Callable
 import cv2
 import gradio as gr
@@ -23,9 +24,11 @@ placeholders = utils.create_placeholder_variables()
 
 @session_logger.set_uuid_logging
 def parse_args(args_to_parse):
+    print(f"ROOT_PROJECT:{utils.ROOT_PROJECT}.")
+    logging.info(f"ROOT_PROJECT:{utils.ROOT_PROJECT}.")
     parser = argparse.ArgumentParser(description="LISA chat")
     parser.add_argument("--version", default="xinlai/LISA-13B-llama2-v1-explanatory")
-    parser.add_argument("--vis_save_path", default="./vis_output", type=str)
+    parser.add_argument("--vis_save_path", default=str(Path(utils.ROOT_PROJECT) / "vis_output"), type=str)
     parser.add_argument(
         "--precision",
         default="fp16",
@@ -333,3 +336,8 @@ def get_gradio_interface(
         examples=constants.examples,
         allow_flagging="auto"
     )
+
+
+if __name__ == '__main__':
+    arrrrg = parse_args([])
+    print("arrrrg:", arrrrg)

lisa_on_cuda/utils/utils.py

@@ -1,4 +1,5 @@
 import logging
+import os
 from enum import Enum
 from pathlib import Path
 
@@ -44,6 +45,7 @@ ANSWER_LIST = [
     "[SEG].",
 ]
 ROOT = Path(__file__).parent.parent.parent
+ROOT_PROJECT = os.getenv("ROOT_PROJECT", ROOT)
 
 
 class Summary(Enum):

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "lisa-on-cuda"
-version = "0.1.0"
+version = "0.2.0"
 description = ""
 authors = ["alessandro trinca tornidor <[email protected]>"]
 license = "Apache 2.0"