Update app.py

app.py

@@ -66,8 +66,9 @@ def predict(input_image:Image.Image, true_label:str):
 
  target_layers=[model.block8.branch1[-1]]
  use_cuda = True if torch.cuda.is_available() else False
- '''print ("Cuda :: ", use_cuda)'''
- cam = GradCAM(model=model, target_layers=target_layers) ''', use_cuda=use_cuda)'''
+ #print ("Cuda :: ", use_cuda)
+ cam = GradCAM(model=model, target_layers=target_layers) 
+ #, use_cuda=use_cuda)
  targets = [ClassifierOutputTarget(0)]
 
  grayscale_cam = cam(input_tensor=face, targets=targets, eigen_smooth=True)