conciomith's picture
Update app.py
2b57f68
raw
history blame
306 Bytes
import gradio as gr
import RetinaFace
def RetinaFace(img):
faces = RetinaFace.extract_faces(img, align = True)
return faces[0]
inputs = gr.inputs.Image(shape=(800, 600), source="upload")
gr.Interface(fn=RetinaFace, inputs=inputs, outputs="plot", title="RetinaFace").launch(inbrowser=True)