update

app.py

@@ -2,7 +2,6 @@ import gradio as gr
 from attack import Attacker
 import argparse
 
-
 def do_attack(img, eps, step_size, steps, progress=gr.Progress()):
     args=argparse.Namespace()
     args.out_dir='./'
@@ -19,7 +18,7 @@ def do_attack(img, eps, step_size, steps, progress=gr.Progress()):
 
     attacker = Attacker(args, pgd_callback=pdg_prog)
     atk_img, noise = attacker.attack_(img)
-    attacker.save_image(atk_img, noise, 'out.png')
+    attacker.save_image(img, noise, 'out.png')
     return 'out.png'
 
 with gr.Blocks(title="Anime AI Detect Fucker Demo", theme="dark") as demo: