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)