| import gradio as gr | |
| import subprocess | |
| import os | |
| def videoxsub(vid): | |
| getAudio(vid) | |
| getSubs() | |
| return [vid, "audio.vtt"] | |
| def getAudio(vid): | |
| if os.path.exists("audio.mp3"): | |
| os.remove("audio.mp3") | |
| commands_list = [ "ffmpeg", "-i", vid, "audio.mp3" ] | |
| subprocess.run(commands_list) | |
| def getSubs(): | |
| command_list = [ "whisper", "audio.mp3", "-f", "vtt", "--fp16", "False"] | |
| subprocess.run(command_list) | |
| demo = gr.Interface(fn=videoxsub, inputs="video", outputs="video") | |
| demo.launch() |