conciomith's picture
Update app.py
2a83514
raw
history blame
264 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="image")