import gradio as gr
import cv2

ed     = cv2.ximgproc.createEdgeDrawing()
params = cv2.ximgproc.EdgeDrawing.Params()
params.PFmode = True
ed.setParams(params)

def edpf(image_rgb):
    img_gray = cv2.cvtColor(image_rgb, cv2.COLOR_BGR2GRAY)
    edges    = ed.detectEdges(img_gray)
    edge_map = ed.getEdgeImage(edges)
    return edge_map

demo = gr.Interface(edpf, gr.Image(shape=(512, 512)), "image")

if __name__ == "__main__":
    demo.launch()