diff --git a/README.md b/README.md
index fac88f9..6ac49be 100644
--- a/README.md
+++ b/README.md
@@ -1038,7 +1038,6 @@ Key dependencies include:
 - torchvision (== 0.16.2)
 - diffusers (== 0.29.2)
 - transformers (== 4.42.4)
-- gradio (== 4.44.1)
 - opencv-python (== 4.8.1.78)
 - scikit-image (== 0.22.0)
 - numpy (== 1.26.4)
diff --git a/demo/extract_garment/requirements.txt b/demo/extract_garment/requirements.txt
index 3c886cb..4378aa8 100644
--- a/demo/extract_garment/requirements.txt
+++ b/demo/extract_garment/requirements.txt
@@ -1,3 +1,3 @@
-gradio==4.44.1
+gradio==6.0.1
 pillow
 tryondiffusion
\ No newline at end of file
diff --git a/demo/model_swap/app.py b/demo/model_swap/app.py
index 2013922..7ddae45 100644
--- a/demo/model_swap/app.py
+++ b/demo/model_swap/app.py
@@ -227,7 +227,7 @@ def rerender1(user_token1):
                     label="Token",
                     lines=1,
                     value=user_token1,
-                    type="password",
+                    password=True,
                     placeholder="Enter your access token here!",
                     info="Visit https://playground.tryonlabs.ai to retrieve your access token."
                 )
diff --git a/demo/model_swap/requirements.txt b/demo/model_swap/requirements.txt
index 33cf3e6..669c964 100644
--- a/demo/model_swap/requirements.txt
+++ b/demo/model_swap/requirements.txt
@@ -1,2 +1,2 @@
-gradio==4.44.1
-gradio_modal==0.0.3
\ No newline at end of file
+gradio>=4.0,<6.0
+gradio_modal==0.0.4
\ No newline at end of file
diff --git a/demo/outfit_generator/requirements.txt b/demo/outfit_generator/requirements.txt
index eefc24b..1165c36 100644
--- a/demo/outfit_generator/requirements.txt
+++ b/demo/outfit_generator/requirements.txt
@@ -1,5 +1,5 @@
 spaces
-gradio
+gradio==6.0.1
 diffusers
 torch
 numpy
diff --git a/environment.yml b/environment.yml
index a308a15..0d524c7 100644
--- a/environment.yml
+++ b/environment.yml
@@ -63,8 +63,6 @@ dependencies:
     - cycler==0.11.0
     - debugpy==1.8.0
     - decorator==5.1.1
-    - diffusers==0.29.2
-    - einops==0.7.0
     - exceptiongroup==1.1.3
     - executing==1.2.0
     - fastapi==0.108.0
@@ -77,9 +75,6 @@ dependencies:
     - google-auth==2.23.3
     - google-auth-oauthlib==1.0.0
     - google-pasta==0.2.0
-    - gradio==4.39.0
-    - gradio-client==1.1.1
-    - gradio-modal==0.0.3
     - h11==0.14.0
     - h5py==3.10.0
     - httpcore==1.0.5
@@ -102,8 +97,6 @@ dependencies:
     - markdown==3.5
     - markdown-it-py==3.0.0
     - markupsafe==2.1.3
-    - matplotlib==3.8.0
-    - matplotlib-inline==0.1.6
     - mdurl==0.1.2
     - ml-dtypes==0.2.0
     - nest-asyncio==1.5.8
@@ -142,7 +135,6 @@ dependencies:
     - ruff==0.5.5
     - safetensors==0.4.3
     - scikit-image==0.22.0
-    - scipy==1.11.4
     - semantic-version==2.10.0
     - setuptools==69.0.3
     - shellingham==1.5.4
diff --git a/requirements.txt b/requirements.txt
index f265dc7..29014f3 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -3,17 +3,10 @@ torchvision==0.16.2
 numpy==1.26.4
 opencv-python==4.8.1.78
 pillow==10.1.0
-matplotlib==3.8.0
 tqdm==4.66.1
-einops==0.7.0
-scipy==1.11.4
 scikit-image==0.22.0
 python-dotenv==1.0.1
-diffusers==0.29.2
 transformers==4.42.4
-gradio==4.44.1
-gradio-modal==0.0.3
-python-dotenv
 boto3==1.40.64
 requests>=2.31.0
 PyJWT>=2.10.1
diff --git a/setup.py b/setup.py
index a49df4c..4e3e02a 100644
--- a/setup.py
+++ b/setup.py
@@ -23,16 +23,10 @@
         "numpy==1.26.4",
         "opencv-python==4.8.1.78",
         "pillow==10.1.0",
-        "matplotlib==3.8.0",
         "tqdm==4.66.1",
-        "einops==0.7.0",
-        "scipy==1.11.4",
         "scikit-image==0.22.0",
         "python-dotenv==1.0.1",
-        "diffusers==0.29.2",
         "transformers==4.42.4",
-        "gradio==4.44.1",
-        "gradio-modal==0.0.3",
         "boto3==1.40.64",
         "requests>=2.31.0",
         "PyJWT>=2.10.1",