victorisgeek commited on
Commit
43f8d6f
·
verified ·
1 Parent(s): 54fb304

Delete app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -89
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)