Spaces:
Running
on
A10G
Running
on
A10G
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() | |