Update handler.py

handler.py

@@ -40,7 +40,7 @@ uvicorn_thread.start()
 
 authtoken = "2cvqFKWc1eb9b0aN7pRLDUBfEtC_2FUehxFL8CAKXRkW3Hfjo"
 commands = [
-    "snap install ngrok",
+    # "snap install ngrok",
     f"ngrok config add-authtoken {authtoken}"
 ]
 for command in commands:
@@ -51,11 +51,6 @@ for command in commands:
         logger.info(f"Failed CMD: {e}")
 
 
-subprocess.Popen(['ngrok', 'http', '5000'], stdout=subprocess.PIPE, stderr=subprocess.PIPE)
-
-logger.info("Ngrok is running in the background.")
-
-
 class EndpointHandler:
     def __init__(self, path=""):
         self.pipe = FluxPipeline.from_pretrained(
@@ -71,6 +66,9 @@ class EndpointHandler:
         )
         # ngrok
         self.public_url = ngrok.connect(5000).public_url
+        subprocess.Popen(['ngrok', 'http', '5000'], stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+        
+        logger.info("Ngrok is running in the background.")
 
         
     def __call__(self, data: Dict[str, Any]) -> str: