add debug mode

app.py

@@ -103,7 +103,7 @@ MAX_SEED = np.iinfo(np.int32).max
 TRIGGER = "a photo of TOK"
 
 @spaces.GPU(duration=100)
-def execute(image, prompt):
+def execute(image, prompt, debug=False):
   if not prompt :
         gr.Info("Please enter a text prompt.")
         return None
@@ -146,6 +146,10 @@ def execute(image, prompt):
             max_sequence_length=256,
             joint_attention_kwargs={"scale": 0.9},
         ).images[0]
+
+    if debug:
+        response.appeend(im)
+        response.appeend(mask)
     response.append(result)
 
   return response
@@ -160,7 +164,8 @@ iface = gr.Interface(
     title=title,
     inputs=[
         gr.Image(type="filepath"),
-        gr.Textbox(label="Prompt")
+        gr.Textbox(label="Prompt"),
+        gr.Checkbox(label="Debug Mode")
     ],
     outputs="gallery"
 )