Update handler.py
Browse files- handler.py +2 -2
handler.py
CHANGED
|
@@ -65,8 +65,8 @@ class EndpointHandler:
|
|
| 65 |
self.pipe.vae, mode="max-autotune-no-cudagraphs",
|
| 66 |
)
|
| 67 |
# ngrok
|
| 68 |
-
self.public_url = ngrok.connect(
|
| 69 |
-
subprocess.Popen(['ngrok', 'http', '
|
| 70 |
|
| 71 |
logger.info("Ngrok is running in the background.")
|
| 72 |
|
|
|
|
| 65 |
self.pipe.vae, mode="max-autotune-no-cudagraphs",
|
| 66 |
)
|
| 67 |
# ngrok
|
| 68 |
+
self.public_url = ngrok.connect(6000).public_url
|
| 69 |
+
subprocess.Popen(['ngrok', 'http', '6000'], stdout=subprocess.PIPE, stderr=subprocess.PIPE)
|
| 70 |
|
| 71 |
logger.info("Ngrok is running in the background.")
|
| 72 |
|