import gradio as gr

def monai_inference(input):
    pass

demo = gr.Interface(
    fn=monai_inference,
    inputs=gr.File(file_count="single", file_types=[".nii.gz", ".zip"], label=".nii.gz file or .zip file"),
    outputs="file",
    title="Inference on monai model",
    description="You can upload either zip of dicom folder or .nii.gz file. In turn, you can download the mask as .nii.gz file.",
)

demo.launch()