feat: added transcribe func

app.py

@@ -1,7 +1,19 @@
 import gradio as gr
+import pytube
+import whisper
 
-def greet(name):
-    return "Hello " + name + "!!"
 
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
+
+def get_audio(video_url):
+    # Download YouTube video
+    yt = pytube.YouTube(video_url)
+    return yt.streams.first(only_audio=True).download(filename="tmpvideo.mp4")
+
+
+def transcribe(video_url):
+    # Transcribe audio
+    model = whisper.load_model("base")
+    return model.transcribe(get_audio(video_url))
+
+iface = gr.Interface(fn=transcribe, inputs="text", outputs="text")
 iface.launch()

requirements.txt

@@ -0,0 +1,3 @@
+transformers
+pytube
+git+https://github.com/openai/whisper.git