Update app.py

app.py

@@ -35,7 +35,7 @@ def get_new_prompt(img, mode):
 def infer(prompt, mode):
     img = get_images(prompt) 
     result = get_new_prompt(img, mode)    
-    return result[0]
+    return result[0], img
 
 with gr.Blocks() as demo:
     
@@ -45,11 +45,12 @@ with gr.Blocks() as demo:
             prompt_input = gr.Textbox(lines=4, label="Input v1.x Stable Diffusion prompt")
             mode_input = gr.Radio(['best', 'classic', 'fast'], label='mode', value='fast')
             submit_btn = gr.Button("Submit")
-        
-        prompt_output = gr.Textbox(lines=4, label="Converted v2.x Stable Diffusion prompt")
+        with gr.Column():
+            sd_inter = gr.Image()
+            prompt_output = gr.Textbox(lines=4, label="Converted v2.x Stable Diffusion prompt")
 
     submit_btn.click(
-        fn=infer, inputs=[prompt_input,mode_input], outputs=[prompt_output]
+        fn=infer, inputs=[prompt_input,mode_input], outputs=[prompt_output, sd_inter]
     )
 
     examples=[
@@ -62,7 +63,7 @@ with gr.Blocks() as demo:
         examples = examples,
         fn = infer,
         inputs=[prompt_input,mode_input],
-        outputs=[prompt_output]
+        outputs=[prompt_output, sd_inter]
     )
 
 #demo=gr.Interface(fn=infer, inputs=[prompt_input,mode_input], outputs=[prompt_output],title=title,description=description,examples=examples)