dzy7e commited on
Commit
d459a94
1 Parent(s): 7dd5168
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -19,7 +19,7 @@ def do_attack(img, eps, step_size, steps, progress=gr.Progress()):
19
  attacker = Attacker(args, pgd_callback=pdg_prog)
20
  atk_img, noise = attacker.attack_(img)
21
  attacker.save_image(img, noise, 'out.png')
22
- return 'out.png'
23
 
24
  with gr.Blocks(title="Anime AI Detect Fucker Demo", theme="dark") as demo:
25
  gr.HTML('<a href="https://github.com/7eu7d7/anime-ai-detect-fucker">github repo</a>')
 
19
  attacker = Attacker(args, pgd_callback=pdg_prog)
20
  atk_img, noise = attacker.attack_(img)
21
  attacker.save_image(img, noise, 'out.png')
22
+ return 'out_atk.png'
23
 
24
  with gr.Blocks(title="Anime AI Detect Fucker Demo", theme="dark") as demo:
25
  gr.HTML('<a href="https://github.com/7eu7d7/anime-ai-detect-fucker">github repo</a>')