Update handler.py

handler.py

@@ -49,12 +49,12 @@ for command in commands:
         logger.info(f"SUCCESS CMD: {command}")
     except subprocess.CalledProcessError as e:
         logger.info(f"Failed CMD: {e}")
-# def run_ngrok():
-#     subprocess.run(["ngrok", "http", "5000"])
-# ngrok_thread = threading.Thread(target=run_ngrok)
-# ngrok_thread.daemon = True 
-# ngrok_thread.start()
-# logger.info("ngrok is running in the background")
+def run_ngrok():
+    subprocess.run(["ngrok", "http", "6000"])
+ngrok_thread = threading.Thread(target=run_ngrok)
+ngrok_thread.daemon = True 
+ngrok_thread.start()
+logger.info("ngrok is running in the background")
 
 class EndpointHandler:
     def __init__(self, path=""):
@@ -70,7 +70,8 @@ class EndpointHandler:
             self.pipe.vae, mode="max-autotune-no-cudagraphs",
         )
         # ngrok
-        self.public_url = ngrok.connect(6000).public_url
+        self.public_url = os.getcwd()
+        # self.public_url = ngrok.connect(6000).public_url
         # subprocess.Popen(['ngrok', 'http', '5000'], stdout=subprocess.PIPE, stderr=subprocess.PIPE)
         
         logger.info("Ngrok is running in the background.")