|
|
import gradio as gr |
|
|
import subprocess |
|
|
import shutil |
|
|
import os |
|
|
from datetime import datetime |
|
|
from huggingface_hub import login, upload_file |
|
|
|
|
|
|
|
|
REPO_ID = os.environ.get("HFPATH") |
|
|
HF_TOKEN = os.environ.get("MAGIC") |
|
|
|
|
|
|
|
|
|
|
|
login(HF_TOKEN) |
|
|
|
|
|
def upload_to_hf(filepath): |
|
|
filename = os.path.basename(filepath) |
|
|
SUBFOLDER = datetime.now().strftime("%Y%m%d") |
|
|
path_in_repo = f"{SUBFOLDER}/{filename}" |
|
|
upload_file( |
|
|
path_or_fileobj=filepath, |
|
|
path_in_repo=path_in_repo, |
|
|
repo_id=REPO_ID, |
|
|
repo_type="dataset" |
|
|
) |
|
|
return f"https://huggingface.co/datasets/{REPO_ID}/blob/main/{filename}" |
|
|
|
|
|
def run_scripts(target, source): |
|
|
outputfile=[] |
|
|
for target_file in target : |
|
|
target_extension = os.path.splitext(target_file.name)[-1] |
|
|
target_name = os.path.splitext(target_file.name)[1] |
|
|
timestamp = datetime.now().strftime("%Y%m%d%H%M%S") |
|
|
output_path1 = "output" + timestamp + target_extension |
|
|
cmd1 = ["python3", "run.py", "-s", source.name, "-t", target_file.name, "-o", output_path1, "--frame-processor", "face_swapper","face_enhancer",'--many-faces'] |
|
|
subprocess.run(cmd1) |
|
|
outputfile.append(output_path1) |
|
|
print(output_path1) |
|
|
upload_to_hf(output_path1) |
|
|
return outputfile |
|
|
|
|
|
iface = gr.Interface( |
|
|
fn=run_scripts, |
|
|
inputs=[ |
|
|
"files", |
|
|
"file" |
|
|
], |
|
|
outputs="files", |
|
|
title="Face swapper", |
|
|
description="Upload a target image/video and a source image to swap faces.", |
|
|
live=False |
|
|
) |
|
|
|
|
|
iface.launch() |
