diff --git a/fashn_viton.py b/fashn_viton.py
index d3e050b..6309172 100644
--- a/fashn_viton.py
+++ b/fashn_viton.py
@@ -18,12 +18,12 @@ class FASHN:
             "required": {
                 "model_image": ("IMAGE",),
                 "garment_image": ("IMAGE",),
-                "category": (["tops", "bottoms", "one-pieces"],),
             },
             "optional": {
+                "category": (["tops", "bottoms", "one-pieces", "auto"], {"default": "auto"}),
                 "mode": (["performance", "balanced", "quality"], {"default": "balanced"}),
                 "garment_photo_type": (["auto", "model", "flat-lay"], {"default": "auto"}),
-                "nsfw_filter": ("BOOLEAN", {"default": True}),
+                "moderation_level": (["none", "permissive", "conservative"], {"default": "permissive"}),
                 "cover_feet": ("BOOLEAN", {"default": False}),
                 "adjust_hands": ("BOOLEAN", {"default": False}),
                 "restore_background": ("BOOLEAN", {"default": False}),
@@ -101,7 +101,7 @@ class FASHN:
         category,
         mode,
         garment_photo_type,
-        nsfw_filter,
+        moderation_level,
         cover_feet,
         adjust_hands,
         restore_background,
@@ -146,7 +146,7 @@ class FASHN:
             "category": category,
             "mode": mode,
             "garment_photo_type": garment_photo_type,
-            "nsfw_filter": nsfw_filter,
+            "moderation_level": moderation_level,
             "cover_feet": cover_feet,
             "adjust_hands": adjust_hands,
             "restore_background": restore_background,