Added app.py

app.py

@@ -0,0 +1,81 @@
+import gradio as gr
+from pytube import YouTube
+from transformers import pipeline
+import whisper
+
+class GradioInference():
+  def __init__(self):
+    self.model = pipeline(model="humeur/lab2_id2223")
+    self.yt = None
+  
+  def __call__(self, link, subs):
+    if self.yt is None:
+      self.yt = YouTube(link)
+    path = self.yt.streams.filter(only_audio=True)[0].download(filename="tmp.mp4")
+
+    results = self.loaded_model(path)
+
+    if subs == "None":
+      return results["text"]
+    elif subs == ".srt":
+      return self.srt(results["segments"])
+    elif ".csv" == ".csv":
+      return self.csv(results["segments"])
+
+  def srt(self, segments):
+    output = ""
+    for i, segment in enumerate(segments):
+      output += f"{i+1}\n"
+      output += f"{self.format_time(segment['start'])} --> {self.format_time(segment['end'])}\n"
+      output += f"{segment['text']}\n\n"
+    return output
+
+  def csv(self, segments):
+    output = ""
+    for segment in segments:
+      output += f"{segment['start']},{segment['end']},{segment['text']}\n"
+    return output
+
+  def format_time(self, time):
+    hours = time//3600
+    minutes = (time - hours*3600)//60
+    seconds = time - hours*3600 - minutes*60
+    milliseconds = (time - int(time))*1000
+    return f"{int(hours):02d}:{int(minutes):02d}:{int(seconds):02d},{int(milliseconds):03d}"
+
+  def populate_metadata(self, link):
+    self.yt = YouTube(link)
+    return self.yt.thumbnail_url, self.yt.title
+
+gio = GradioInference()
+title="SWED->EN Youtube Transcriber (Whisper)"
+description="Speech to text transcription of Youtube videos using OpenAI's Whisper finetunned for Swedish to English translation"
+
+block = gr.Blocks()
+with block:
+    gr.HTML(
+        f"""
+            <div style="text-align: center; max-width: 500px; margin: 0 auto;">
+              <div>
+                <h1>{title}</h1>
+              </div>
+              <p style="margin-bottom: 10px; font-size: 94%">
+                {description}
+              </p>
+            </div>
+        """
+    )
+    with gr.Group():
+        with gr.Box():
+          with gr.Row().style(equal_height=True):
+            wt = gr.Radio(["No", ".srt", ".csv"], label="Result with timestamps?")
+          link = gr.Textbox(label="YouTube Link")
+          title = gr.Label(label="Video Title")
+          with gr.Row().style(equal_height=True):
+            img = gr.Image(label="Thumbnail")
+            text = gr.Textbox(label="Transcription", placeholder="Transcription Output", lines=10)
+          with gr.Row().style(equal_height=True):
+              btn = gr.Button("Transcribe")
+          btn.click(gio, inputs=[link, wt], outputs=[text])
+          link.change(gio.populate_metadata, inputs=[link], outputs=[img, title])
+block.launch()