Update app.py

app.py

@@ -64,7 +64,7 @@ def inference(
         )
 
         # return replace_nsfw_images(result)
-        return result.images, results.nsfw_content_detected, f"Done. Seed: {seed}"
+        return result.images, results.nsfw_content_detected, seed
     except Exception as e:
         return None, error_str(e)
 
@@ -121,7 +121,8 @@ with gr.Blocks(css="style.css") as demo:
             state_info = gr.Textbox(label="State", show_label=False, max_lines=2).style(
                 container=False
             )
-            error_output = gr.Markdown()
+            nsfw_output = gr.JSON()
+            error_output = gr.JSON()
 
         with gr.Column(scale=45):
             with gr.Tab("Options"):
@@ -182,7 +183,7 @@ with gr.Blocks(css="style.css") as demo:
         height,
         seed,
     ]
-    outputs = [gallery, error_output]
+    outputs = [gallery, nsfw_output, error_output]
     prompt.submit(inference, inputs=inputs, outputs=outputs)
     generate.click(inference, inputs=inputs, outputs=outputs)