File size: 1,675 Bytes
111e796 81a551f 111e796 84168d6 111e796 1b676fc |
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 49 50 51 52 53 54 55 56 57 58 59 60 61 62 |
import gradio as gr
import whisper
from whisper.utils import write_vtt
import requests
import os
import re
model = whisper.load_model("medium")
def inference(link):
content = requests.get(link)
podcast_url = re.findall("(?P<url>\;https?://[^\s]+)", content.text)[0].split(';')[1]
print(podcast_url)
download = requests.get(podcast_url)
with open('podcast.mp3', 'wb') as f:
f.write(download.content)
result = model.transcribe('podcast.mp3')
with open('sub.vtt', "w") as txt:
write_vtt(result["segments"], file=txt)
return (result['text'], 'sub.vtt')
title="PodScript"
description="Get Podcast Transcript"
block = gr.Blocks()
with block:
gr.HTML(
""" <center>
<h1>PodScript</h1>
<img src = 'https://external-content.duckduckgo.com/iu/?u=https%3A%2F%2Fwww.newbreedmarketing.com%2Fhs-fs%2Fhubfs%2Fshutterstock_1125707303.jpg%3Fwidth%3D5000%26name%3Dshutterstock_1125707303.jpg&f=1&nofb=1&ipt=0ba3d9b639d63b0b737ea63cd81b241cc47f46cb519ac4ca18fd3ce6fc1376ad&ipo=images' width = '20%'></img>
</center>
"""
)
with gr.Group():
with gr.Box():
link = gr.Textbox(label="Google Podcasts Link")
with gr.Row().style(mobile_collapse=False, equal_height=True):
btn = gr.Button("Get PodScript 🪄")
text = gr.Textbox(
label="PodScript",
placeholder="PodScript Output",
lines=5)
file = gr.File()
btn.click(inference, inputs=[link], outputs=[text,file])
block.launch(debug=True, enable_queue = True)
|