yangcsu/facialdetection-vgg

1import gradio as gr2import os3from src.detection import Detector4 5 6# UGC: Define the inference fn() for your models7def model_inference(image):8    image, json_out = Detector('BlazeFace')(image)9    return image10 11 12def clear_all():13    return None, None, None14 15 16# 下载模型17os.system("wget -c https://huggingface.co/yangcsu/facialdetection-vgg/resolve/main/vgg.pdparams -P ./configs")18os.system("wget -c https://huggingface.co/yangcsu/facialdetection-vgg/resolve/main/model.pdiparams -P ./configs")19os.system("wget -c https://huggingface.co/yangcsu/facialdetection-vgg/resolve/main/model.pdmodel -P ./configs")20 21examples = [22    "images/1.jpg",23    "images/2.jpg",24    "images/3.jpg",25]26title = "BlazeFace+VGG的人脸表情分析"27 28demo = gr.Interface(fn=model_inference, inputs="image", outputs="image", title=title,29                    examples=examples)30 31# 启动Gradio32demo.launch()33