Update app.py

app.py

@@ -130,11 +130,11 @@ with gr.Blocks(css=custom_css) as demo:
  start_button = gr.Button("Attack prepare!",size='lg')
  text_input = gr.Textbox(label="Input Prompt")
  
- orig_img = gr.Image(label="Image Generated by Input Prompt",width=512,show_share_button=False,show_download_button=False)
+ orig_img = gr.Image(label="Image Gnerated by Unlearned DM using input prompt",width=512,show_share_button=False,show_download_button=False)
  with gr.Column():
  attack_button = gr.Button("UnlearnDiffAtk!",size='lg')
  text_ouput = gr.Textbox(label="Prompt Genetated by UnlearnDiffAtk")
- result_img = gr.Image(label="Image Gnerated by Prompt of UnlearnDiffAtk",width=512,show_share_button=False,show_download_button=False)
+ result_img = gr.Image(label="Image Gnerated by Unlearned DM using adversarial prompt",width=512,show_share_button=False,show_download_button=False)
  
  start_button.click(fn=execute_prepare, inputs=[drop_model, drop, shown_columns_step, atk_idx], outputs=[text_input, orig_img], api_name="prepare")
  attack_button.click(fn=execute_udiff, inputs=[drop_model, drop, shown_columns_step, atk_idx], outputs=[text_ouput, result_img], api_name="udiff")