Update app.py

app.py

@@ -141,7 +141,7 @@ weights = "weights_nm.h5"
 def get_grad(img):
     img = img_pros(img)
     grad_model = create_grad_model(weights, 2, "softmax")
-    grad_imgs, y_pred, infer = gen_grad_both(grad_model, img)
+    grad_img_c, grad_img_d, y_pred, infer = gen_grad_both(grad_model, img)
     # pred_class = ""
     # if y_pred[0] > 0.5: pred_class = "cat"
     # else: pred_class = "dog"