Update test.py

xh-liu · Aug 27, 2021 · f2101d7 · f2101d7
1 parent 820b347
commit f2101d7
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/test.py b/test.py
@@ -99,7 +99,8 @@
     print(message)
 
 # manipulation results after optimizing perturbations
-visuals['optimized'] = open_edit_optimizer.get_latest_generated()[0]
+manipulated_opt = open_edit_optimizer.open_edit_model(data, mode='manipulate', ori_cap=ori_txt, new_cap=new_txt, alpha=alpha)
+visuals['optimized_manipulated'] = manipulated_opt[0]
 
 
 visualizer.save_images(webpage, visuals, [opt.img_path], gray=True)