Update app.py

app.py

@@ -35,7 +35,8 @@ def get_new_prompt(img, mode):
 def infer(prompt, mode):
     img = get_images(prompt) 
     result = get_new_prompt(img, mode)    
-    return result[0], img
+    #return result[0], img
+    return img
 
 with gr.Blocks() as demo:
     
@@ -50,7 +51,7 @@ with gr.Blocks() as demo:
             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, sd_inter]
+        fn=infer, inputs=[prompt_input,mode_input], outputs=[sd_inter]
     )
 
     examples=[
@@ -63,7 +64,7 @@ with gr.Blocks() as demo:
         examples = examples,
         fn = infer,
         inputs=[prompt_input,mode_input],
-        outputs=[prompt_output, sd_inter]
+        outputs=[sd_inter]
     )
 
 #demo=gr.Interface(fn=infer, inputs=[prompt_input,mode_input], outputs=[prompt_output],title=title,description=description,examples=examples)