import gradio as gr
from main import main

iface = gr.Interface(
        fn=main,
        inputs=gr.Image(type="filepath", label="Invoice Image"),
        outputs=gr.Image(type="pil", label="Masked Invoice Image"),
        live=True
    )

iface.launch()