From 348421dd6d2fc50fd2a5f7eaab303bc0e598302c Mon Sep 17 00:00:00 2001
From: Sean Sube <seansube@gmail.com>
Date: Sat, 22 Apr 2023 11:26:21 -0500
Subject: [PATCH] fix controlnet segmentation string, add some source filter
 logging

---
 api/onnx_web/diffusers/run.py | 1 +
 api/onnx_web/server/api.py    | 1 +
 gui/src/strings/en.ts         | 2 +-
 3 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/api/onnx_web/diffusers/run.py b/api/onnx_web/diffusers/run.py
index fe0943b8..6c5ce55b 100644
--- a/api/onnx_web/diffusers/run.py
+++ b/api/onnx_web/diffusers/run.py
@@ -348,6 +348,7 @@ def run_img2img_pipeline(
     if source_filter is not None:
         f = get_source_filters().get(source_filter, None)
         if f is not None:
+            logger.debug("running source filter: %s", f.__name__)
             source = f(server, source)
 
     pipe = load_pipeline(
diff --git a/api/onnx_web/server/api.py b/api/onnx_web/server/api.py
index 109b0a66..fe662f79 100644
--- a/api/onnx_web/server/api.py
+++ b/api/onnx_web/server/api.py
@@ -179,6 +179,7 @@ def img2img(server: ServerContext, pool: DevicePoolExecutor):
 
     output_count = params.batch
     if source_filter is not None:
+        logger.debug("including filtered source with outputs, filter: %s", source_filter)
         output_count += 1
 
     output = make_output_name(
diff --git a/gui/src/strings/en.ts b/gui/src/strings/en.ts
index 466b3ee9..469ab11f 100644
--- a/gui/src/strings/en.ts
+++ b/gui/src/strings/en.ts
@@ -121,7 +121,7 @@ export const I18N_STRINGS_EN = {
           normal: 'Normal Map',
           openpose: 'OpenPose',
           scribble: 'Scribble',
-          segment: 'Image Segmentation',
+          seg: 'Image Segmentation',
         },
       },
       modelType: {