fix: annotator preview issue (Mikubill#496)

ch8os · Mar 5, 2023 · 63f7d3c · 63f7d3c
1 parent eab5bc0
commit 63f7d3c
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/scripts/controlnet.py b/scripts/controlnet.py
@@ -420,10 +420,12 @@ def run_annotator(image, module, pres, pthr_a, pthr_b):
             preprocessor = self.preprocessor[module]
             result = None
             if pres > 64:
-                result = preprocessor(img, res=pres, thr_a=pthr_a, thr_b=pthr_b)
+                result, is_image = preprocessor(img, res=pres, thr_a=pthr_a, thr_b=pthr_b)
             else:
-                result = preprocessor(img)
-            return gr.update(value=result, visible=True, interactive=False)
+                result, is_image = preprocessor(img)
+
+            if is_image:
+                return gr.update(value=result, visible=True, interactive=False)
 
         with gr.Row():
             annotator_button = gr.Button(value="Preview annotator result")