diff --git a/api/dataset_dict.pkl b/api/dataset_dict_30.pkl similarity index 100% rename from api/dataset_dict.pkl rename to api/dataset_dict_30.pkl diff --git a/api/get_blocks.py b/api/get_blocks.py index 54a78bb..f1c939c 100644 --- a/api/get_blocks.py +++ b/api/get_blocks.py @@ -14,7 +14,7 @@ COMPLETIONS_MODEL = "gpt-3.5-turbo" import pathlib project_path = pathlib.Path(__file__).parent -PATH_TO_DATASET_DICT = project_path / "dataset_dict.pkl" +PATH_TO_DATASET_DICT = project_path / "dataset_dict_500.pkl" with open(PATH_TO_DATASET_DICT, 'rb') as f: data = pickle.load(f)