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()