Update app.py

app.py

@@ -2,15 +2,21 @@ import gradio as gr
 import whisper
 from pytube import YouTube
 
+pipe = pipeline(model="Mei000/whisper-small-sv-SE")
 
 class GradioInference():
   def __init__(self):
     self.yt = None
-  
-  def __call__(self, link, lang, size, subs):
+    self.loaded_model = pipe
+      
+  def __call__(self, link, subs):
     if self.yt is None:
-      self.yt = YouTube(link)
+       self.yt = YouTube(link)
     path = self.yt.streams.filter(only_audio=True)[0].download(filename="tmp.mp4")
+    results = self.loaded_model.transcribe(path)
+
+    if subs == "None":
+      return results["text"]
     
   def populate_metadata(self, link):
     self.yt = YouTube(link)