|
import gradio as gr |
|
from gradio_client import Client |
|
|
|
sdxl_client = Client("https://fffiloni-sdxl-dpo.hf.space/") |
|
faceswap_client = Client("https://tonyassi-face-swap.hf.space/") |
|
|
|
def infer(portrait_in, prompt_in): |
|
|
|
gr.Info("Generating SDXL image first ...") |
|
sdxl_result = sdxl_client.predict( |
|
prompt_in, |
|
api_name="/infer" |
|
) |
|
|
|
|
|
gr.Info("Face swap your face on result ...") |
|
faceswap_result = faceswap_client.predict( |
|
portrait_in, |
|
sdxl_result, |
|
api_name="/predict" |
|
) |
|
|
|
return faceswap_result |
|
|
|
css = """ |
|
#col-container{ |
|
margin: 0 auto; |
|
max-width: 840px; |
|
} |
|
""" |
|
with gr.Blocks(css=css) as demo: |
|
with gr.Column(elem_id="col-container"): |
|
gr.HTML(""" |
|
<h2 style="text-align: center;">Portrait Maker</h2> |
|
""") |
|
with gr.Row(): |
|
with gr.Column(): |
|
portrait_in = gr.Image(label="Your face portrait", type="filepath") |
|
prompt_in = gr.Textbox(label="Prompt to desired portrait using your own face") |
|
submit_btn = gr.Button("Submit") |
|
with gr.Column(): |
|
result = gr.Image(label="Result") |
|
|
|
submit_btn.click( |
|
fn = infer, |
|
inputs = [ |
|
portrait_in, |
|
prompt_in |
|
], |
|
outputs = [ |
|
result |
|
] |
|
) |
|
|
|
demo.queue().launch() |
