diff --git a/src/dataset/create_dataset.py b/src/dataset/create_dataset.py index 516e4cb..bdf5182 100644 --- a/src/dataset/create_dataset.py +++ b/src/dataset/create_dataset.py @@ -21,7 +21,7 @@ except ImportError: openai.api_key = os.environ.get('OPENAI_API_KEY') -from .settings import PATH_TO_DATASET_PKL, PATH_TO_DATASET_DICT_PKL, EMBEDDING_MODEL, LEN_EMBEDDINGS +from .settings import PATH_TO_DATASET_DICT_PKL, EMBEDDING_MODEL, LEN_EMBEDDINGS from .text_splitter import TokenSplitter, split_into_sentences diff --git a/src/main.py b/src/main.py index 333795a..029df5a 100644 --- a/src/main.py +++ b/src/main.py @@ -27,7 +27,7 @@ if str(src_path) not in sys.path: from dataset import create_dataset #from assistant import semantic_search -from settings import EMBEDDING_MODEL, PATH_TO_DATASET_DICT_PKL, PATH_TO_DATASET_PKL +from settings import EMBEDDING_MODEL, PATH_TO_DATASET_DICT_PKL import numpy as np import matplotlib.pyplot as plt