Update app.py

app.py

@@ -78,7 +78,9 @@ def send_it2(inputs, noise_level, proc1=proc1):
     #output4 = proc1(prompt_with_noise)
     #return output4
 
-
+def identity_with_sleep(x):
+    time.sleep(0.5)
+    return x
 
 with gr.Blocks(css='style.css') as demo:
     gr.HTML(
@@ -137,6 +139,12 @@ with gr.Blocks(css='style.css') as demo:
         see_prompts.click(get_prompts, inputs=[input_text], outputs=[prompt], queue=False)
         run.click(send_it1, inputs=[prompt, noise_level], outputs=[output1])
         run.click(send_it2, inputs=[prompt, noise_level], outputs=[output2])
+
+        with gr.Column():
+            input_img = gr.Image()
+            output_img = gr.Image()
+            submit_img = gr.Button()
+            submit_img.click(identity_with_sleep, input_img, output_img, api_name="img")
         
 
 
@@ -153,7 +161,6 @@ with gr.Blocks(css='style.css') as demo:
         </div>
     """
 )
-    btn.click(add, [num1, num2], output, api_name="addition")
 
     demo.launch(enable_queue=True, inline=True)
     block.queue(concurrency_count=100)