Update app.py

app.py

@@ -131,7 +131,7 @@ def gen_grad_both(grad_model, img):
     # 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)
     
     infer = ""
-    if y_pred_c[0] > y_pred_c[1]: infer = "cat"
+    if y_pred_c[0][0] > y_pred_c[0][1]: infer = "cat"
     else: infer = "dog"
         
     return img_c, img_d, y_pred_c, infer