VGG-Face / app.py
fugam's picture
Create app.py
0416c72 verified
Raw
History Blame Contribute Delete
802 Bytes
import gradio as gr
from deepface import DeepFace
# Charger le modèle VGG-Face avec ses poids
model = DeepFace.build_model("VGG-Face")
# Fonction pour comparer deux images de visages
def compare_faces(image1, image2):
# Utiliser DeepFace pour vérifier si les deux images correspondent
result = DeepFace.verify(image1, image2, model_name="VGG-Face")
return result
# Interface Gradio pour télécharger et comparer les images
iface = gr.Interface(
fn=compare_faces,
inputs=[gr.Image(type="filepath"), gr.Image(type="filepath")],
outputs="json",
title="Reconnaissance Faciale avec VGG-Face",
description="Téléchargez deux images de visages pour vérifier si elles sont similaires. Le modèle utilisé est VGG-Face."
)
# Lancer l'interface Gradio
iface.launch()