admin/LC_NeoRefacer
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Added types to arg and follows the server port from arg

Browse Source 
Added types to the argument to prevent errors when parsing.

also makes the ngrok use ports from the server_port argument
This commit is contained in:
Alunite
2023-06-08 20:23:43 +08:00
parent 0613ce21e6
commit 3e13367f85
1 changed files with 8 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
app.py
View File

@@ -4,14 +4,14 @@ import argparse
import ngrok
parser = argparse.ArgumentParser(description='Refacer')
parser.add_argument("--max_num_faces", help="Max number of faces on UI", default=5)
parser.add_argument("--force_cpu", help="Force CPU mode", default=False,action="store_true")
parser.add_argument("--share_gradio", help="Share Gradio", default=False,action="store_true")
parser.add_argument("--server_name", help="Server IP address", default="127.0.0.1")
parser.add_argument("--server_port", help="Server port", type=int, default=7860)
parser.add_argument("--max_num_faces", type=int, help="Max number of faces on UI", default=5)
parser.add_argument("--force_cpu", help="Force CPU mode", default=False, action="store_true")
parser.add_argument("--share_gradio", help="Share Gradio", default=False, action="store_true")
parser.add_argument("--server_name", type=str, help="Server IP address", default="127.0.0.1")
parser.add_argument("--server_port", type=int, help="Server port", default=7860)
parser.add_argument("--colab_performance", help="Use in colab for better performance", default=False,action="store_true")
parser.add_argument("--ngrok", help="Use ngrok", default=None)
parser.add_argument("--ngrok_region", help="ngrok region", default="us")
parser.add_argument("--ngrok", type=str, help="Use ngrok", default=None)
parser.add_argument("--ngrok_region", type=str, help="ngrok region", default="us")
args = parser.parse_args()
refacer = Refacer(force_cpu=args.force_cpu,colab_performance=args.colab_performance)
@@ -87,7 +87,7 @@ with gr.Blocks() as demo:
button.click(fn=run,inputs=[video]+origin+destination+thresholds,outputs=[video2])
if args.ngrok is not None:
connect(args.ngrok, 7860, {'region': args.ngrok_region, 'authtoken_from_env': False})
connect(args.ngrok, args.server_port, {'region': args.ngrok_region, 'authtoken_from_env': False})
#demo.launch(share=True,server_name="0.0.0.0", show_error=True)
demo.queue().launch(show_error=True,share=args.share_gradio,server_name=args.server_name,server_port=args.server_port)