fffiloni's picture
Create app.py
34caf6a
raw
history blame
No virus
1.46 kB
import gradio as gr
from gradio_client import Client
sdxl_client = Client("https://fffiloni-sdxl-dpo.hf.space/--replicas/8llck/")
faceswap_client = Client("https://tonyassi-face-swap.hf.space/")
def infer(portrait_in, prompt_in):
# Generate Image from SDXL
gr.Info("Generating SDXL image first ...")
sdxl_result = sdxl_client.predict(
prompt_in,
api_name="/infer"
)
# Face Swap
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()