Update app.py

app.py

@@ -4,6 +4,7 @@ from pytube import YouTube
 from fastapi import FastAPI, Response, Request
 import yt_dlp
 import uvicorn
+import os
 
 CUSTOM_PATH = "/gradio"
 
@@ -38,6 +39,8 @@ def get_subtitle(url, lang='en'):
         video_id = info_dict.get("id", None)
         if video_id is None:
             return None
+         os.listdir()
+        print(info_dict)
         subtitle_file = f"{video_id}.{lang}.vtt"
         with open(subtitle_file, 'r') as f:
             subtitle_content = f.read()
@@ -75,14 +78,16 @@ def get_audio(url):
     return yt.streams.filter(only_audio=True)[0].download(filename="tmp.mp4")
 
 def get_transcript(url, model_size, lang, format):
+    if lang == "None":
+        lang = None
+
     subtitle = get_subtitle(url, lang)
     print(subtitle)
     return subtitle
 
     model = whisper.load_model(model_size)
 
-    if lang == "None":
-        lang = None
+    
     
     result = model.transcribe(download_audio(url), fp16=False, language=lang)