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

Merge pull request #26 from alunit3/main

Browse Source 
Added type annotation to prevent parsing errors and makes the ngrok use ports from the server_port argument
This commit is contained in:
xaviviro
2023-06-08 15:16:17 +02:00
committed by GitHub
parent a51089ea84 92f1b28e8d
commit 61decb67a6
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)