Spaces:
Running
Running
File size: 2,241 Bytes
c0e5ce3 326cf1a c0e5ce3 cbd8427 c0e5ce3 326cf1a cbd8427 326cf1a cbd8427 c0e5ce3 326cf1a 5c87c18 326cf1a 5c87c18 326cf1a 5c87c18 c0e5ce3 326cf1a c0e5ce3 326cf1a c0e5ce3 cbd8427 d1aeee7 cbd8427 c0e5ce3 34976a0 |
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 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 |
import gradio as gr
from prodiapy import Prodia
from PIL import Image
from io import BytesIO
import requests
import random
import os
import base64
client = Prodia()
def infer(source, target):
if source_image is None or target_image is None:
return
source_url = upload_image(source)
target_url = upload_image(target)
job = client.faceswap(source_url=source_url, target_url=target_url)
res = client.wait(job, raise_on_fail=False)
if res.failed:
return
return res.image_url
def upload_image(file):
files = {'file': open(file, 'rb')}
img_id = requests.post(os.getenv("IMAGE_API_1"), files=files).json()['id']
payload = {
"content": "",
"nonce": f"{random.randint(1, 10000000)}H9X42KSEJFNNH",
"replies": [],
"attachments": [img_id]
}
res = requests.post(os.getenv("IMAGE_API_2"), json=payload, headers={"x-session-token": os.getenv("SESSION_TOKEN")})
return f"{os.getenv('IMAGE_API_1')}/{img_id}/{res.json()['attachments'][0]['filename']}"
def image_to_base64(image: Image):
# Convert the image to bytes
buffered = BytesIO()
image.save(buffered, format="PNG") # You can change format to PNG if needed
# Encode the bytes to base64
img_str = base64.b64encode(buffered.getvalue())
return img_str.decode('utf-8') # Convert bytes to string
with gr.Blocks() as demo:
with gr.Column():
gr.HTML("<h1><center>Face Swap</center></h1>")
with gr.Row():
with gr.Row():
source_image = gr.Image(type="filepath", label="Source Image")
target_image = gr.Image(type="filepath", label="Target Image")
with gr.Column():
result = gr.Image()
run_button = gr.Button("Swap Faces", variant="primary")
gr.Examples(
examples=[
["example1.jpg", "example2.jpg"],
["example3.jpg", "example4.jpg"],
["example5.jpg", "example6.jpg"]
],
fn=infer,
inputs=[source_image, target_image],
outputs=[result]
)
run_button.click(fn=infer, inputs=[source_image, target_image], outputs=[result])
demo.queue(max_size=20, api_open=False).launch(show_api=False, max_threads=400)
|