Resolved PR comments from Boris

2022-10-26 16:30:38 +01:00
parent 401f7c7ef0
commit 209c1a12e8
4 changed files with 20 additions and 30 deletions
--- a/examples/Clustering_for_transaction_classification.ipynb
+++ b/examples/Clustering_for_transaction_classification.ipynb
@ -41,10 +41,9 @@
    "from sklearn.manifold import TSNE\n",
    "import matplotlib\n",
    "import matplotlib.pyplot as plt\n",
+    "import os\n",
    "\n",
-    "from helpers import OPENAI_API_KEY\n",
-    "\n",
-    "openai.api_key = OPENAI_API_KEY\n",
+    "openai.api_key = os.getenv(\"OPENAI_API_KEY\")\n",
    "COMPLETIONS_MODEL = \"text-davinci-002\"\n",
    "\n",
    "# This path leads to a file with embeddings created in the notebook linked above\n",