Update main.py

main.py

@@ -6,6 +6,7 @@ import uuid
 import os
 from fastapi.middleware.cors import CORSMiddleware
 import gradio as gr
+from pyngrok import ngrok
 
 app = FastAPI()
 
@@ -17,6 +18,11 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
+ngrok.set_auth_token("2NtE79QzuHf6t2uV7xiBni14GIx_L5YaQzN96KS61RWS9qi1")
+ngrok.kill()
+ngrok_tunnel = ngrok.connect(7860)
+print(ngrok_tunnel.public_url)
+
 # or run on GPU with INT8
 # model = WhisperModel(model_size, device="cuda", compute_type="int8_float16")
 # or run on CPU with INT8
@@ -27,6 +33,9 @@ def segment_to_dict(segment):
         segment["words"] = [word._asdict() for word in segment["words"]]
     return segment
 
+def ng_path(text):
+    return ngrok_tunnel.public_url
+
 @app.post("/video")
 async def download_video(video_url: str):
     download_convert_video_to_audio(yt_dlp, video_url, f"/home/user/{uuid.uuid4().hex}")