Update requirements.txt

Sebastian Sosa · Sebastian Sosa · commit e081df0703d7 · 2024-04-16T21:49:10.000-03:00
diff --git a/requirements.txt b/requirements.txt
@@ -1,3 +1,6 @@
+openai==1.19.0
+python-dotenv==1.0.1
 rich==13.7.1
+scikit-learn==1.4.2
 torch==2.2.1
 transformers==4.39.0
diff --git a/research/api_llm_attribution.ipynb b/research/api_llm_attribution.ipynb
@@ -18,7 +18,6 @@
    ],
    "source": [
     "import os\n",
-    "import numpy as np\n",
     "import torch\n",
     "from transformers import AutoTokenizer, AutoModel, PreTrainedModel, PreTrainedTokenizer, GPT2LMHeadModel, GPT2Tokenizer\n",
     "from sklearn.neighbors import NearestNeighbors\n",
