Pix2Pix-Video / app.py
fffiloni's picture
Create app.py
db12400
raw
history blame
696 Bytes
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()