test / app.py
john1246's picture
Update app.py
b5bcc75 verified
raw
history blame
945 Bytes
import os
import subprocess
import gradio as gr
# Clone the GitHub repository
if not os.path.exists("roopapi"):
subprocess.run(["git", "clone", "https://github.com/lvalics/roopapi.git"])
# Change the working directory to the cloned repository
os.chdir("roopapi")
# Install the required dependencies from requirements.txt (if it's not already installed)
subprocess.run(["pip", "install", "-r", "requirements.txt"])
# Now import the ROOP API (assuming `roop.py` is in the cloned repo)
from roop import ROOP
# Initialize ROOP model
roop_model = ROOP()
# Gradio function to swap faces
def process_face_swap(image, video):
output = roop_model.swap_faces(image, video)
return output
# Gradio interface
iface = gr.Interface(
fn=process_face_swap,
inputs=["image", "video"],
outputs="video",
live=True,
title="ROOP Face Swapping API",
description="Upload an image and a video to swap faces."
)
iface.launch()