diff --git a/global_var.py b/global_var.py index 5a449b4..66b3118 100644 --- a/global_var.py +++ b/global_var.py @@ -9,7 +9,7 @@ _model = None _tokenizer = None _workdir = None def init_global_var(workdir="workdir"): - global _prompt_store, _sql_engine, _docs, _datasets + global _prompt_store, _sql_engine, _docs, _datasets, _workdir _prompt_store = get_prompt_tinydb(workdir) _sql_engine = get_sqlite_engine(workdir) _docs = scan_docs_directory(workdir)