Spaces:
Runtime error
Runtime error
from deepface import DeepFace | |
from PIL import Image | |
import requests | |
import tempfile | |
import os | |
def verify_faces(img1_url, img2_url): | |
# Download the images from the URLs | |
response1 = requests.get(img1_url) | |
response2 = requests.get(img2_url) | |
# Check if the requests were successful | |
if response1.status_code == 200 and response2.status_code == 200: | |
# Create temporary files to store the downloaded images | |
with tempfile.NamedTemporaryFile(delete=False, suffix='.jpg') as img1_tempfile: | |
img1_tempfile.write(response1.content) | |
img1_path = img1_tempfile.name | |
with tempfile.NamedTemporaryFile(delete=False, suffix='.jpg') as img2_tempfile: | |
img2_tempfile.write(response2.content) | |
img2_path = img2_tempfile.name | |
# Perform facial recognition with deep_face | |
result = DeepFace.verify(img1_path=img1_path, img2_path=img2_path, model_name='VGG-Face') | |
# Delete the temporary files when done | |
os.remove(img1_path) | |
os.remove(img2_path) | |
if result["verified"]: | |
return "RESULT: Faces Matched!" | |
else: | |
return "RESULT: Faces Don't Match" | |
else: | |
return "Failed to download one or both of the images from the URLs." |