diff --git a/api/gunicorn.conf.py b/api/gunicorn.conf.py deleted file mode 100644 index 2880f19..0000000 --- a/api/gunicorn.conf.py +++ /dev/null @@ -1 +0,0 @@ -TIMEOUT=300 diff --git a/api/main.py b/api/main.py index 332da65..54221f4 100644 --- a/api/main.py +++ b/api/main.py @@ -9,6 +9,7 @@ import pickle import requests # ---------------------------------- env setup --------------------------------- + if os.path.exists('.env'): from dotenv import load_dotenv load_dotenv() @@ -16,7 +17,6 @@ if os.path.exists('.env'): OPENAI_API_KEY = os.environ.get('OPENAI_API_KEY') openai.api_key = OPENAI_API_KEY - # -------------------------------- load dataset -------------------------------- @@ -37,7 +37,7 @@ else: dataset_dict_bytes = requests.get(url).content print('Unpacking dataset...') - dataset_dict = pickle.load(dataset_dict_bytes) + dataset_dict = pickle.loads(dataset_dict_bytes) print('Done!')