Pix2Pix-Video / app.py
fffiloni's picture
Update app.py
bbf9a90
raw
history blame
824 Bytes
import gradio as gr
import os
token = os.environ.get('HF_TOKEN')
pix2pix = gr.Blocks.load(name="spaces/fffiloni/instruct-pix2pix-clone", api_key=token)
def infer(prompt,image):
result = pix2pix(prompt,
5.5,
1.5,
image,
15,
"",
512,
512,
0,
fn_index=0)
print(os.listdir(result))
return [os.path.join(result[0], img) for img in os.listdir(result[0])]
with gr.Blocks() 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_out]
submit_btn.click(infer, inputs, outputs)
demo.launch().queue()