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(RetinaFace, inputs=inputs, outputs="image")