diff --git a/src/main.py b/src/main.py index 99fd3036..4e75537d 100644 --- a/src/main.py +++ b/src/main.py @@ -11,7 +11,7 @@ from PyQt6.QtCore import QThread # Internal Packages from src.configure import configure_server from src.router import router -from src.utils import constants +from src.utils import constants, state from src.utils.cli import cli from src.interface.desktop.configure_screen import ConfigureScreen from src.interface.desktop.system_tray import create_system_tray @@ -25,7 +25,8 @@ app.include_router(router) def run(): # Load config from CLI - args = cli(sys.argv[1:]) + state.cli_args = sys.argv[1:] + args = cli(state.cli_args) # Setup Base GUI gui = QtWidgets.QApplication([]) diff --git a/src/utils/state.py b/src/utils/state.py index 964fa458..90f9296a 100644 --- a/src/utils/state.py +++ b/src/utils/state.py @@ -12,4 +12,5 @@ model = SearchModels() processor_config = ProcessorConfigModel() config_file: Path = "" verbose: int = 0 -device = torch.device("cuda:0") if torch.cuda.is_available() else torch.device("cpu") # Set device to GPU if available \ No newline at end of file +device = torch.device("cuda:0") if torch.cuda.is_available() else torch.device("cpu") # Set device to GPU if available +cli_args = None \ No newline at end of file