Update app.py

app.py

@@ -66,6 +66,7 @@ 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)
     targets = [ClassifierOutputTarget(0)]