Update app.py

app.py

@@ -18,8 +18,6 @@ logging.basicConfig(level=logging.INFO)
 model = ViTForImageClassification.from_pretrained("ongkn/attraction-classifier")
 processor = ViTImageProcessor.from_pretrained("ongkn/attraction-classifier")
 
-pipe = pipeline("image-classification", model=model, feature_extractor=processor)
-
 faceGrabber = FaceGrabber()
 gradCam = GradCam()
 
@@ -50,7 +48,7 @@ def classify_image(input):
                                         input_tensor=tensorResized,
                                         input_image=faceResized,
                                         reshape_transform=gradCam.reshape_transform_vit_huggingface)
-    result = pipe(faceResized)
+    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: