From eb9d4b85a301fdcf97b610680bbbb6bc09ac1cef Mon Sep 17 00:00:00 2001
From: Dan Bochman <dannybochman@gmail.com>
Date: Mon, 23 Dec 2024 15:07:25 +0100
Subject: [PATCH] add mode parameter to FASHN class methods

---
 fashn_viton.py | 2 ++
 1 file changed, 2 insertions(+)

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,