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