import os
import gradio as gr
from utils import recognition

workers = 0 if os.name == 'nt' else 4
device = "cpu"


title = "Compare Faces"
description = "A FaceNet (MTCNN for the bounding boxes and Inception Resnet V1 for the feature extracting, trainning with VGGFace2 dataset) feature extractor computer vision model to classify faces (my family)."
article = "Created by HienK64BKHN."


demo = gr.Interface(fn=recognition,
                    inputs=gr.Image(label="Image"),
                    outputs=[gr.Textbox(label="Names"),
                             gr.Image(label="Result")],
                    title=title,
                    description=description,
                    article=article
                    )

demo.launch(debug=False)