rafat0421 commited on
Commit
bf2e636
·
1 Parent(s): 38dbe36

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -4,6 +4,8 @@ from transformers import pipeline
4
  from pytube import YouTube
5
  from datasets import Dataset, Audio
6
  from moviepy.editor import AudioFileClip
 
 
7
 
8
  pipe = pipeline(model="Neprox/model")
9
 
@@ -70,7 +72,10 @@ def get_translation(text):
70
  Translates the given Swedish text to English.
71
  """
72
  # TODO: Make API call to Google Translate to get English translation
73
- return "..."
 
 
 
74
 
75
  def transcribe(audio, url, seconds_max):
76
  """
 
4
  from pytube import YouTube
5
  from datasets import Dataset, Audio
6
  from moviepy.editor import AudioFileClip
7
+ import googletrans
8
+ from googletrans import Translator
9
 
10
  pipe = pipeline(model="Neprox/model")
11
 
 
72
  Translates the given Swedish text to English.
73
  """
74
  # TODO: Make API call to Google Translate to get English translation
75
+ translator = Translator()
76
+ result = translator.translate(text)
77
+
78
+ return result
79
 
80
  def transcribe(audio, url, seconds_max):
81
  """