Update app.py

app.py

@@ -124,8 +124,8 @@ def gen_grad_img_single(grad_model, img, class_index, alpha = 0.4):
 
 
 def gen_grad_both(grad_model, img):
-    img_c, y_pred_c = gen_grad_img_single(grad_model, imgs, 0)
-    img_d, y_pred_d = gen_grad_img_single(grad_model, imgs, 1)
+    img_c, y_pred_c = gen_grad_img_single(grad_model, img, 0)
+    img_d, y_pred_d = gen_grad_img_single(grad_model, img, 1)
     y_pred_c = np.around(y_pred_c,3)
     y_pred_d = np.around(y_pred_d,3)
     # show_imgs([img_c, img_d], [y_true, y_true], [size[0], size[1]], cols, [y_pred_c, y_pred_d], font_size = font_size)