Spaces:
Running
Running
import gradio as gr | |
import whisper | |
import os | |
model = whisper.load_model("tiny") | |
# main processing | |
def main(URL): | |
video_path = youtube_dl(URL) | |
text = transcript(video_path) | |
return text | |
# Transcribe text using Whisper | |
def transcript(video_path): | |
result = model.transcribe(video_path) | |
return result["text"] | |
# Download the video | |
def youtube_dl(URL): | |
# Download the video | |
os.system(f"yt-dlp -f best -v {URL} -o target.mp4") | |
# Path of downloaded video | |
video_path = os.path.join(os.path.dirname(__file__), "target.mp4") | |
return video_path | |
demo = gr.Interface(fn = main, | |
inputs = "text", outputs = "text") | |
if __name__ == "__main__": | |
demo.launch() | |