Update app.py

app.py

@@ -147,7 +147,7 @@ def get_grad(img):
     # else: pred_class = "dog"
         
     text = "Raw Score: " + str(y_pred[0]) + "\nClassification: " + infer
-    return grad_imgs, text
+    return grad_img_c, grad_img_d, text
 
 demo = gr.Interface(
   fn = get_grad,