Spaces:
Running
Running
File size: 1,703 Bytes
bce23e4 d810840 bce23e4 d810840 bce23e4 9f6db74 bce23e4 d4f3259 342e535 bce23e4 7008ae2 bce23e4 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 |
from os import pipe
import re
import gradio as gr
from functions.punctuation import punctuate
from functions.model_infer import predict_from_document
from functions.convert_time import match_mask_and_transcript
title = "sponsoredBye - never listen to sponsors again"
description = "Sponsored sections in videos are annoying and take up a lot of time. Improve your YouTube watching experience, by filling in the youtube url and figure out what segments to skip."
article = "Check out [the original Rick and Morty Bot](https://huggingface.co/spaces/kingabzpro/Rick_and_Morty_Bot) that this demo is based off of."
def pipeline(video_url):
video_id = video_url.split("?v=")[-1]
punctuated_text, transcript = punctuate(video_id)
sentences = re.split(r"[\.\!\?]\s", punctuated_text)
classification, probs = predict_from_document(sentences)
# return punctuated_text
times, timestamps = match_mask_and_transcript(sentences, transcript, classification)
return [{"begin": time[0], "end": time[1]} for time in times]
# return [
# {
# "start": "12:05",
# "end": "12:52",
# "classification": str(classification),
# "probabilities": probs,
# "times": times,
# "timestamps": timestamps,
# }
# ]
# print(pipeline("VL5M5ZihJK4"))
demo = gr.Interface(
fn=pipeline,
title=title,
description=description,
inputs="text",
# outputs=gr.Label(num_top_classes=3),
outputs="json",
examples=[
"https://www.youtube.com/watch?v=UjtOGPJ0URM",
"https://www.youtube.com/watch?v=TrZyuCh9df0",
],
)
demo.launch(share=True)
|