diff --git a/images/jupyter-ai.yml b/images/jupyter-ai.yml
index 8d83d72..2f482d6 100644
--- a/images/jupyter-ai.yml
+++ b/images/jupyter-ai.yml
@@ -5,7 +5,7 @@ dependencies:
   - gh-scoped-creds
   - ibis-duckdb
   - jupyterlab-myst
-  - jupyter-ai
+    #  - jupyter-ai
   - jupyter-resource-usage
   - jupyter-vscode-proxy
   - pypdf
@@ -13,4 +13,5 @@ dependencies:
   - seaborn
   - pip:
     - jupyter-tensorboard-proxy
+    - jupyter-ai