Update app.py

app.py

@@ -105,12 +105,12 @@ with gr.Blocks(css=custom_css) as demo:
         with gr.Column(min_width=260):
             text_input = gr.Textbox(label="Input Prompt")
             
-            img1 = gr.Image(orig_img,label="Image Generated by Input Prompt",width=260,show_share_button=False,show_download_button=False)
+            orig_img = gr.Image(label="Image Generated by Input Prompt",width=260,show_share_button=False,show_download_button=False)
         with gr.Column():
             start_button = gr.Button("UnlearnDiffAtk!",size='lg')
         with gr.Column(min_width=260):
             text_ouput = gr.Textbox(label="Prompt Genetated by UnlearnDiffAtk")
-            img2 = gr.Image(result_img,label="Image Gnerated by Prompt of UnlearnDiffAtk",width=260,show_share_button=False,show_download_button=False)
+            result_img = gr.Image(abel="Image Gnerated by Prompt of UnlearnDiffAtk",width=260,show_share_button=False,show_download_button=False)
             
 
     start_button.click(fn=excute_udiff, inputs=[drop_model, drop, shown_columns_step], outputs=[text_input, text_ouput, orig_img, result_img], api_name="udiff")