diff --git a/fashn_viton.py b/fashn_viton.py
index 0fc39fd..d3e050b 100644
--- a/fashn_viton.py
+++ b/fashn_viton.py
@@ -99,6 +99,7 @@ class FASHN:
         model_image,
         garment_image,
         category,
+        mode,
         garment_photo_type,
         nsfw_filter,
         cover_feet,
@@ -143,6 +144,7 @@ class FASHN:
             "model_image": model_image,
             "garment_image": garment_image,
             "category": category,
+            "mode": mode,
             "garment_photo_type": garment_photo_type,
             "nsfw_filter": nsfw_filter,
             "cover_feet": cover_feet,