import gradio as gr
import predict_model

def greet(input_img):
    out_img = predict_model.run(input_img)
    return out_img

demo = gr.Interface(greet, inputs=[gr.Image()], outputs=["image"])
demo.launch()