swapmyface / run.py
cosmicweb98's picture
Update run.py
2703678 verified
raw
history blame contribute delete
724 Bytes
#!/usr/bin/env python3
from roop import core
if __name__ == '__main__':
core.run()
import gradio as gr
import cv2
import numpy as np
def face_swap(target_video, swap_image):
# Your face swap logic here
result_video_path = "path/to/result/video.mp4"
return result_video_path
iface = gr.Interface(
fn=face_swap,
inputs=[
gr.Video(label="Target Video", max_size=500*1024*1024), # 500 MB limit
gr.Image(label="Swap Image", type="filepath", max_size=100*1024*1024) # 100 MB limit
],
outputs=gr.Video(label="Result Video"),
title="Face Swap",
description="Upload a target video and a swap image to perform a face swap.",
enable_api=True
)
iface.launch(share=True)