Update app.py

app.py

@@ -42,13 +42,16 @@ def classify_image(input):
                                 n_components=5,
                                 top_k=10
                                 )
+    result = gradCam.get_top_category(model, tensorResized)
+    cls = result[0]["label"]
+    clsIdx = gradCam.category_name_to_index(model, cls)
+    clsTarget = ClassifierOutputTarget(clsIdx)
     gradCamImage = gradCam.run_grad_cam_on_image(model=model,
                                         target_layer=targetLayerGradCam,
-                                        targets_for_gradcam=targetsForGradCam,
+                                        targets_for_gradcam=[clsTarget],
                                         input_tensor=tensorResized,
                                         input_image=faceResized,
                                         reshape_transform=gradCam.reshape_transform_vit_huggingface)
-    result = gradCam.get_top_category(model, tensorResized)
     if result[0]["label"] == "pos" and result[0]["score"] > 0.9 and result[0]["score"] < 0.95:
         return result[0]["label"], result[0]["score"], str("Nice!"), face, dffImage, gradCamImage
     elif result[0]["label"] == "pos" and result[0]["score"] > 0.95: