humeur commited on
Commit
fed269d
1 Parent(s): 3882a86
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ class GradioInference():
14
  path = self.yt.streams.filter(only_audio=True)[0].download(filename="tmp.mp4")
15
  results = self.transcribe_model(path)
16
  results = self.translate_model(results["text"])
17
- return results['translation_text']
18
 
19
  def populate_metadata(self, link):
20
  self.yt = YouTube(link)
 
14
  path = self.yt.streams.filter(only_audio=True)[0].download(filename="tmp.mp4")
15
  results = self.transcribe_model(path)
16
  results = self.translate_model(results["text"])
17
+ return results[0]['translation_text']
18
 
19
  def populate_metadata(self, link):
20
  self.yt = YouTube(link)