HienK64BKHN's picture
Upload 2 files
7a62482 verified
raw
history blame
773 Bytes
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 images of food as pizza, steak or sushi."
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)