Просмотр исходного кода

Add listening port options for listening mode.

Tensa 3 лет назад
Родитель
Сommit
3742d3b18a
1 измененных файлов с 2 добавлено и 1 удалено
  1. 2 1
      server.py

+ 2 - 1
server.py

@@ -36,6 +36,7 @@ parser.add_argument('--no-stream', action='store_true', help='Don\'t stream the
 parser.add_argument('--settings', type=str, help='Load the default interface settings from this json file. See settings-template.json for an example.')
 parser.add_argument('--extensions', type=str, help='The list of extensions to load. If you want to load more than one extension, write the names separated by commas and between quotation marks, "like,this".')
 parser.add_argument('--listen', action='store_true', help='Make the web UI reachable from your local network.')
+parser.add_argument('--listen-port', type=int, choices=range(0, 65535), help='The listening port the server will use.')
 parser.add_argument('--share', action='store_true', help='Create a public URL. This is useful for running the web UI on Google Colab or similar.')
 parser.add_argument('--verbose', action='store_true', help='Print the prompts to the terminal.')
 args = parser.parse_args()
@@ -647,6 +648,6 @@ else:
 
 interface.queue()
 if args.listen:
-    interface.launch(share=args.share, server_name="0.0.0.0")
+    interface.launch(share=args.share, server_name="0.0.0.0", server_port=args.listen_port)
 else:
     interface.launch(share=args.share)