facial-auth-api / app.py
huggingfacesrc's picture
Update app.py
dfff45b verified
Raw History Blame Contribute Delete
800 Bytes
from fastapi import FastAPI, UploadFile, File
from deepface import DeepFace
import cv2
import numpy as np
import io
import uvicorn
app = FastAPI()
REFERENCE_IMAGE_PATH = "reference.jpg" # Add this manually in Hugging Face Space
@app.post("/verify-face/")
async def verify_face(file: UploadFile = File(...)):
try:
contents = await file.read()
image_array = np.frombuffer(contents, np.uint8)
img = cv2.imdecode(image_array, cv2.IMREAD_COLOR)
result = DeepFace.verify(img1_path=img, img2_path=REFERENCE_IMAGE_PATH, model_name="VGG-Face")
return {"verified": result["verified"], "distance": result["distance"]}
except Exception as e:
return {"error": str(e)}
if __name__ == __main__():
uvicorn.run(app , host = "0.0.0.0" , port = 8080)