Update app.py

app.py

@@ -6,6 +6,8 @@ import cv2 as cv
 import dlib
 import logging
 from typing import Optional
+import os
+
 
 logging.basicConfig(level=logging.INFO)
 
@@ -65,12 +67,22 @@ def grab_faces(img: np.ndarray) -> Optional[np.ndarray]:
 
     return None
 
+def increment_visit():
+    if not os.path.exists("visits.txt"):
+        with open("visits.txt", "w") as f:
+            f.write("0")
+    with open("visits.txt", "r") as f:
+        visits = int(f.read())
+        f.seek(0)
+        f.write(str(visits + 1))
+
 model = ViTForImageClassification.from_pretrained("ongkn/attraction-classifier")
 processor = ViTImageProcessor.from_pretrained("ongkn/attraction-classifier")
 
 pipe = pipeline("image-classification", model=model, feature_extractor=processor)
 
 def classify_image(input):
+    increment_visit() # this is just to keep track of how many people are using this app. no personal data is obviously used
     face = grab_faces(np.array(input))
     if face is None:
         return "No face detected", 0, input