diff --git a/app.py b/app.py index 69a5b82..15221f7 100644 --- a/app.py +++ b/app.py @@ -7,6 +7,8 @@ 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("--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") @@ -88,4 +90,4 @@ if args.ngrok is not None: connect(args.ngrok, 7860, {'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) \ No newline at end of file +demo.queue().launch(show_error=True,share=args.share_gradio,server_name=args.server_name,server_port=args.server_port)