Spaces:
Paused
Paused
Delete app.py
Browse files
app.py
DELETED
|
@@ -1,89 +0,0 @@
|
|
| 1 |
-
import gradio as gr
|
| 2 |
-
from refacer import Refacer
|
| 3 |
-
import ngrok
|
| 4 |
-
import time
|
| 5 |
-
import threading
|
| 6 |
-
|
| 7 |
-
# Initialize Refacer
|
| 8 |
-
def initialize_refacer():
|
| 9 |
-
global refacer
|
| 10 |
-
print("Initializing Refacer...")
|
| 11 |
-
start_time = time.time()
|
| 12 |
-
refacer = Refacer(force_cpu=False, colab_performance=False)
|
| 13 |
-
print(f"Refacer initialized in {time.time() - start_time:.2f} seconds")
|
| 14 |
-
|
| 15 |
-
# Ngrok connection
|
| 16 |
-
def connect(token, port, options):
|
| 17 |
-
account = None
|
| 18 |
-
if token:
|
| 19 |
-
if ':' in token:
|
| 20 |
-
# token = authtoken:username:password
|
| 21 |
-
token, username, password = token.split(':', 2)
|
| 22 |
-
account = f"{username}:{password}"
|
| 23 |
-
|
| 24 |
-
if not options.get('authtoken_from_env'):
|
| 25 |
-
options['authtoken'] = token
|
| 26 |
-
if account:
|
| 27 |
-
options['basic_auth'] = account
|
| 28 |
-
|
| 29 |
-
try:
|
| 30 |
-
public_url = ngrok.connect(f"127.0.0.1:{port}", **options).url()
|
| 31 |
-
print(f'ngrok connected to localhost:{port}! URL: {public_url}')
|
| 32 |
-
except Exception as e:
|
| 33 |
-
print(f'ngrok connection aborted: {e}')
|
| 34 |
-
|
| 35 |
-
# Run reface
|
| 36 |
-
def run(*vars):
|
| 37 |
-
video_path = vars[0]
|
| 38 |
-
origins = vars[1:(num_faces + 1)]
|
| 39 |
-
destinations = vars[(num_faces + 1):(num_faces * 2) + 1]
|
| 40 |
-
thresholds = vars[(num_faces * 2) + 1:]
|
| 41 |
-
|
| 42 |
-
faces = []
|
| 43 |
-
for k in range(num_faces):
|
| 44 |
-
if origins[k] and destinations[k]:
|
| 45 |
-
faces.append({
|
| 46 |
-
'origin': origins[k],
|
| 47 |
-
'destination': destinations[k],
|
| 48 |
-
'threshold': thresholds[k]
|
| 49 |
-
})
|
| 50 |
-
|
| 51 |
-
return refacer.reface(video_path, faces)
|
| 52 |
-
|
| 53 |
-
# UI setup
|
| 54 |
-
origin = []
|
| 55 |
-
destination = []
|
| 56 |
-
thresholds = []
|
| 57 |
-
num_faces = 5
|
| 58 |
-
|
| 59 |
-
with gr.Blocks() as demo:
|
| 60 |
-
with gr.Row():
|
| 61 |
-
gr.Markdown("# Refacer")
|
| 62 |
-
with gr.Row():
|
| 63 |
-
video = gr.Video(label="Original video", format="mp4")
|
| 64 |
-
video2 = gr.Video(label="Refaced video", interactive=False, format="mp4")
|
| 65 |
-
|
| 66 |
-
for i in range(num_faces):
|
| 67 |
-
with gr.Tab(f"Face #{i + 1}"):
|
| 68 |
-
with gr.Row():
|
| 69 |
-
origin.append(gr.Image(label="Face to replace"))
|
| 70 |
-
destination.append(gr.Image(label="Destination face"))
|
| 71 |
-
with gr.Row():
|
| 72 |
-
thresholds.append(gr.Slider(label="Threshold", minimum=0.0, maximum=1.0, value=0.2))
|
| 73 |
-
|
| 74 |
-
with gr.Row():
|
| 75 |
-
button = gr.Button("Reface", variant="primary")
|
| 76 |
-
|
| 77 |
-
button.click(fn=run, inputs=[video] + origin + destination + thresholds, outputs=[video2])
|
| 78 |
-
|
| 79 |
-
# Start initialization in a separate thread to prevent blocking
|
| 80 |
-
init_thread = threading.Thread(target=initialize_refacer)
|
| 81 |
-
init_thread.start()
|
| 82 |
-
|
| 83 |
-
# Launch demo
|
| 84 |
-
demo = gr.Interface(fn=predict, inputs=gr.inputs.Textbox(label='CANDIDATE input'), \
|
| 85 |
-
outputs=[gr.outputs.Textbox(label='VACANCIES'),\
|
| 86 |
-
gr.Plot()],\
|
| 87 |
-
css='div {margin-left: auto; margin-right: auto; width: 100%;\
|
| 88 |
-
background-image: url("https://ibb.co/37wzh0r"); repeat 0 0;}')\
|
| 89 |
-
.launch(share=False).launch(show_error=True, share=True, server_name="0.0.0.0", server_port=7860)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|