diff --git a/environment.yaml b/environment.yaml
index e32bc9b..04af769 100644
--- a/environment.yaml
+++ b/environment.yaml
@@ -1,4 +1,4 @@
-name: ldm
+name: ldm2
 # This file is part of sygil-webui (https://github.com/Sygil-Dev/sygil-webui/).
 
 # Copyright 2022 Sygil-Dev team.
@@ -17,12 +17,12 @@ name: ldm
 channels:
   - pytorch
   - defaults
+  - nvidia
 # Psst. If you change a dependency, make sure it's mirrored in the docker requirement
 # files as well.
 dependencies:
   - nodejs=18.11.0
   - yarn=1.22.19
-  - create-docusaurus=2.2.0
   - cudatoolkit=11.3
   - git
   - numpy=1.22.3
diff --git a/package.json b/package.json
index d090a0a..4bcfcfe 100644
--- a/package.json
+++ b/package.json
@@ -19,6 +19,7 @@
     "@docusaurus/preset-classic": "2.2.0",
     "@mdx-js/react": "^1.6.22",
     "clsx": "^1.2.1",
+    "create-docusaurus": "^2.2.0",
     "prism-react-renderer": "^1.3.5",
     "react": "^17.0.2",
     "react-dom": "^17.0.2"
diff --git a/requirements.txt b/requirements.txt
index 3a0c8a5..e57a46c 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -33,7 +33,7 @@ streamlit-server-state==0.14.2
 streamlit-tensorboard==0.0.2
 streamlit-elements==0.1.* # used for the draggable dashboard and new UI design (WIP)
 streamlit-ace==0.1.1 # used to replace the text area on the prompt and also for the code editor tool.
-streamlit-base-extras # used for logging, thread spawning, thread locking and page routing.
+#streamlit-base-extras # used for logging, thread spawning, thread locking and page routing. For now we are using a modified local version which we have to change to use the proper version from pypi.
 hydralit==1.0.14
 hydralit_components==1.0.10
 stqdm==0.0.4
@@ -44,6 +44,8 @@ matplotlib==3.6.
 resize-right==0.0.2
 torchdiffeq==0.2.3
 
+# Environment Dependencies for WebUI (flet)
+
 # txt2vid
 diffusers==0.6.0
 librosa==0.9.2