Download app.py from Rockk08/FaceRecognition: direct link, hf CLI and curl.
- Browser
- Download file 1.78 kB
-
https://huggingface.co/Rockk08/FaceRecognition/resolve/main/app.py
- Command line
-
hf download hf://Rockk08/FaceRecognition/app.py
-
curl -L -o app.py https://huggingface.co/Rockk08/FaceRecognition/resolve/main/app.py
1.78 kB
| import requests | |
| import pyrebase | |
| import urllib | |
| from retinaface import RetinaFace | |
| from deepface import DeepFace | |
| from fastapi import FastAPI | |
| app = FastAPI() | |
| def read_root(): | |
| return {"Hello": "World"} | |
| def verify_image(url1): | |
| firebaseConfig = { | |
| "apiKey": "AIzaSyClnRJAnrJgAgkYjuYnlvu-CJ6Cxyklebo", | |
| "authDomain": "socioverse-2025.firebaseapp.com", | |
| "projectId": "socioverse-2025", | |
| "storageBucket": "socioverse-2025.appspot.com", | |
| "messagingSenderId": "689574504641", | |
| "appId": "1:689574504641:web:a22f6a2fa343e4221acc40", | |
| "databaseURL": "https://console.firebase.google.com/project/socioverse-2025/storage/socioverse-2025.appspot.com/files", | |
| "serviceAccount":"Firebase_Service_Account_Keys.json" | |
| }; | |
| firebase = pyrebase.initialize_app(firebaseConfig) | |
| storage = firebase.storage() | |
| path = "Faces/" | |
| files = storage.bucket.list_blobs(prefix=path) | |
| flag = False | |
| # url1 = "https://api.time.com/wp-content/uploads/2023/04/shah-rukh-khan-time100-2023-1.jpg" | |
| for file in files: | |
| if file.name.endswith(".jpg"): | |
| # print(file.name) | |
| url = storage.child(f"{file.name}").get_url(None) | |
| # print(url) | |
| with requests.get(url) as response: | |
| result = DeepFace.verify(f"{url1}",url, model_name="Facenet", distance_metric='cosine') | |
| if result['verified']: | |
| flag = True | |
| start_index = file.name.rfind('/') | |
| end_index = file.name.rfind('$') | |
| if start_index != -1 and end_index != -1: | |
| name = file.name[start_index + 1:end_index] | |
| return {"username": name} | |
| break | |
| if flag == False: | |
| print("Not Verified") | |
| return {"username": "Not Found"} | |