diff --git a/requirements.txt b/requirements.txt
index 536ba0f2..0e056bdc 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -36,17 +36,13 @@ opencv-python
 
 prettytable
 safetensors
-streamlit
+google-generativeai
 
 test-tube
 timm
-# torchmetrics
-
 webdataset
 yapf
-wolframalpha
 
-wikipedia==1.4.0
 httpx
 ggl
 
diff --git a/setup.py b/setup.py
index 26b32560..19f9f999 100644
--- a/setup.py
+++ b/setup.py
@@ -28,6 +28,7 @@ setup(
         'nest_asyncio',
         'bs4',
         'pegasusx',
+        'google-generativeai',
         'oceandb',
         'codeinterpreterapi',
         'playwright',