Spaces:
Runtime error
Runtime error
from pytube import YouTube | |
import ffmpeg | |
from gradio_client import Client | |
import subprocess | |
import gradio as gr | |
client = Client("abidlabs/Acapellify") | |
placeholder_url = "https://www.youtube.com/watch?v=bfATETsvqt4&ab_channel=MiniMuslims" | |
placeholder_file = "example_acapella_video.mp4" | |
def acapellify(url): | |
if url == placeholder_url: | |
return placeholder_file | |
yt = YouTube(url) | |
try: | |
video = yt.streams.get_highest_resolution() | |
video.download(max_retries=5) | |
except: | |
raise gr.Error("YouTube video could not be downloaded. Try clicking Submit again and make sure that the URL is correct.") | |
video_path = video.get_file_path() | |
( | |
ffmpeg | |
.input(video_path) | |
.audio | |
.output('audio.m4a', format='mp4') | |
.overwrite_output() | |
.run() | |
) | |
result = client.predict( | |
'audio.m4a', | |
api_name="/predict" | |
) | |
subprocess.call(['ffmpeg', '-y', '-i', video_path, '-i', result[0], '-map', '0:v', '-map', '1:a', '-c:v', 'copy', '-c:a', 'aac', '-strict', 'experimental', 'acapella_video.mp4']) | |
return "acapella_video.mp4" | |
io = gr.Interface(acapellify, gr.Textbox(label="URL to YouTube video", value=placeholder_url), gr.Video(label="Acapella output")) | |
with gr.Blocks() as demo: | |
gr.Markdown("# Acapellify: Get a Vocals-only from any YouTube Video in 1 Click") | |
with gr.Row(): | |
with gr.Column(): | |
gr.Markdown("**Example Original Video**") | |
gr.Video("old_video.mp4") | |
with gr.Column(): | |
gr.Markdown("**Example Acapella Video**") | |
gr.Video("acapella_video.mp4") | |
gr.Markdown("**Try it yourself:** (for now, I recommend short videos <3 minutes)") | |
io.render() | |
demo.launch() |