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]