test

app.py

@@ -53,6 +53,7 @@ def predict(inp):
   cam = GradCAM(model=model,
        target_layers=target_layers)
   img = ho_trans_center(image = inp)['image']
+  rgb_img = img.copy()
   img = topt(image = img)['image']
   img = img.unsqueeze(0)
   with torch.no_grad():
@@ -67,7 +68,7 @@ def predict(inp):
   grad = grad[0 :]
   print(img.shape, img.dtype)
   print(grad.shape, grad.dtype)
-  cam_image = show_cam_on_image(img, grad)
+  cam_image = show_cam_on_image(rgb_img, grad)
   return confidences,cam_image 
 
 import gradio as gr