aboutsummaryrefslogtreecommitdiffstats
path: root/modules/ngrok.py
diff options
context:
space:
mode:
authorddPn08 <pyn.g.1125@gmail.com>2022-10-16 00:24:01 +0000
committerAUTOMATIC1111 <16777216c@gmail.com>2022-10-16 06:56:33 +0000
commit3395ba493f93214cf037d084d45693a37610bd85 (patch)
tree03a3bafb74eb3fe893fc76f4660bedf6bba3e6f7 /modules/ngrok.py
parent179e3ca752d0133470fd3ae44153ee0b71450c9f (diff)
downloadstable-diffusion-webui-gfx803-3395ba493f93214cf037d084d45693a37610bd85.tar.gz
stable-diffusion-webui-gfx803-3395ba493f93214cf037d084d45693a37610bd85.tar.bz2
stable-diffusion-webui-gfx803-3395ba493f93214cf037d084d45693a37610bd85.zip
Allow specifying the region of ngrok.
Diffstat (limited to 'modules/ngrok.py')
-rw-r--r--modules/ngrok.py8
1 files changed, 5 insertions, 3 deletions
diff --git a/modules/ngrok.py b/modules/ngrok.py
index 7d03a6df..5c5f349a 100644
--- a/modules/ngrok.py
+++ b/modules/ngrok.py
@@ -1,12 +1,14 @@
from pyngrok import ngrok, conf, exception
-def connect(token, port):
+def connect(token, port, region):
if token == None:
token = 'None'
- conf.get_default().auth_token = token
+ config = conf.PyngrokConfig(
+ auth_token=token, region=region
+ )
try:
- public_url = ngrok.connect(port).public_url
+ public_url = ngrok.connect(port, pyngrok_config=config).public_url
except exception.PyngrokNgrokError:
print(f'Invalid ngrok authtoken, ngrok connection aborted.\n'
f'Your token: {token}, get the right one on https://dashboard.ngrok.com/get-started/your-authtoken')