Update app.py

app.py

@@ -6,9 +6,13 @@ from google.oauth2 import service_account
 from googleapiclient.discovery import build
 from googleapiclient.http import MediaIoBaseDownload, MediaFileUpload
 from PIL import Image
-import torch
 import io
-from deepface import DeepFace  # Use DeepFace for advanced face recognition
+import numpy as np
+from deepface import DeepFace
+from scipy.spatial.distance import cosine  # Import cosine similarity calculation
+
+# Disable GPU to avoid cuDNN and cuBLAS errors
+os.environ["CUDA_VISIBLE_DEVICES"] = "-1"
 
 # Load Google Drive API credentials
 SERVICE_ACCOUNT_FILE = './.env'  # Assuming the uploaded JSON file is named '.env'
@@ -73,8 +77,9 @@ def extract_face_embedding(image_path):
 
 def calculate_similarity(encoding1, encoding2):
     # Use cosine similarity for comparison
-    similarity = DeepFace.find(img_path1=encoding1, img_path2=encoding2, distance_metric='cosine')[0]
-    return 1 - similarity  # Convert distance to similarity
+    distance = cosine(encoding1, encoding2)
+    similarity = 1 - distance
+    return similarity
 
 def compare_faces_with_encodings(encoding_cctv, aadhar_encodings, threshold=0.7):
     for aadhar_name, aadhar_encoding in aadhar_encodings: