import gradio as gr

pix2pix = gr.Blocks.load(name="spaces/patrickvonplaten/instruct-pix2pix")

def infer(prompt,image):
    result = pix2pix(prompt,
        5.5,
        1.5,
        image,
        15,
        "",
        512,
        512,
        0,
        fn_index=0)

    print(result)
    return result[0]

with gr.Block() as demo:
    with gr.Column():
        prompt = gr.Textbox(placeholder="enter prompt")
        image = gr.Image(source="upload", type="filepath")
        gallery_out = gr.Gallery()
        submit_btn = gr.Button("go")

    inputs = [
        prompt,
        image
    ]
    outputs = [gallery]
    
    submit_btn.click(infer, inputs, outputs)

demo.launch().queue()