Ii commited on
Commit
ca04c9a
β€’
1 Parent(s): 060b7a0

Delete app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -93
app.py DELETED
@@ -1,93 +0,0 @@
1
- import gradio as gr
2
- from refacer import Refacer
3
- import argparse
4
- import ngrok
5
-
6
- parser = argparse.ArgumentParser(description='Refacer')
7
- parser.add_argument("--max_num_faces", type=int, help="Max number of faces on UI", default=5)
8
- parser.add_argument("--force_cpu", help="Force CPU mode", default=False, action="store_true")
9
- parser.add_argument("--share_gradio", help="Share Gradio", default=False, action="store_true")
10
- parser.add_argument("--server_name", type=str, help="Server IP address", default="127.0.0.1")
11
- parser.add_argument("--server_port", type=int, help="Server port", default=7860)
12
- parser.add_argument("--colab_performance", help="Use in colab for better performance", default=False,action="store_true")
13
- parser.add_argument("--ngrok", type=str, help="Use ngrok", default=None)
14
- parser.add_argument("--ngrok_region", type=str, help="ngrok region", default="us")
15
- args = parser.parse_args()
16
-
17
- refacer = Refacer(force_cpu=args.force_cpu,colab_performance=args.colab_performance)
18
-
19
- num_faces=args.max_num_faces
20
-
21
- # Connect to ngrok for ingress
22
- def connect(token, port, options):
23
- account = None
24
- if token is None:
25
- token = 'None'
26
- else:
27
- if ':' in token:
28
- # token = authtoken:username:password
29
- token, username, password = token.split(':', 2)
30
- account = f"{username}:{password}"
31
-
32
- # For all options see: https://github.com/ngrok/ngrok-py/blob/main/examples/ngrok-connect-full.py
33
- if not options.get('authtoken_from_env'):
34
- options['authtoken'] = token
35
- if account:
36
- options['basic_auth'] = account
37
-
38
-
39
- try:
40
- public_url = ngrok.connect(f"127.0.0.1:{port}", **options).url()
41
- except Exception as e:
42
- print(f'Invalid ngrok authtoken? ngrok connection aborted due to: {e}\n'
43
- f'Your token: {token}, get the right one on https://dashboard.ngrok.com/get-started/your-authtoken')
44
- else:
45
- print(f'ngrok connected to localhost:{port}! URL: {public_url}\n'
46
- 'You can use this link after the launch is complete.')
47
-
48
-
49
- def run(*vars):
50
- video_path=vars[0]
51
- origins=vars[1:(num_faces+1)]
52
- destinations=vars[(num_faces+1):(num_faces*2)+1]
53
- thresholds=vars[(num_faces*2)+1:]
54
-
55
- faces = []
56
- for k in range(0,num_faces):
57
- if origins[k] is not None and destinations[k] is not None:
58
- faces.append({
59
- 'origin':origins[k],
60
- 'destination':destinations[k],
61
- 'threshold':thresholds[k]
62
- })
63
-
64
- return refacer.reface(video_path,faces)
65
-
66
- origin = []
67
- destination = []
68
- thresholds = []
69
-
70
- with gr.Blocks() as demo:
71
- with gr.Row():
72
- gr.Markdown("# Refacer")
73
- with gr.Row():
74
- video=gr.Video(label="Original video",format="mp4")
75
- video2=gr.Video(label="Refaced video",interactive=False,format="mp4")
76
-
77
- for i in range(0,num_faces):
78
- with gr.Tab(f"Face #{i+1}"):
79
- with gr.Row():
80
- origin.append(gr.Image(label="Face to replace"))
81
- destination.append(gr.Image(label="Destination face"))
82
- with gr.Row():
83
- thresholds.append(gr.Slider(label="Threshold",minimum=0.0,maximum=1.0,value=0.2))
84
- with gr.Row():
85
- button=gr.Button("Reface", variant="primary")
86
-
87
- button.click(fn=run,inputs=[video]+origin+destination+thresholds,outputs=[video2])
88
-
89
- if args.ngrok is not None:
90
- connect(args.ngrok, args.server_port, {'region': args.ngrok_region, 'authtoken_from_env': False})
91
-
92
- #demo.launch(share=True,server_name="0.0.0.0", show_error=True)
93
- demo.queue().launch(show_error=True,share=args.share_gradio,server_name=args.server_name,server_port=args.server_port)