Spaces:
Running
on
A10G
Running
on
A10G
File size: 824 Bytes
db12400 b127fd6 bbf9a90 db12400 3954375 fe6d862 db12400 0d8ee1e db12400 56a7f5f db12400 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 |
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()
|