Neprox commited on
Commit
0babf4c
·
1 Parent(s): 4cae6cd
Files changed (2) hide show
  1. app.py +5 -4
  2. requirements.txt +1 -1
app.py CHANGED
@@ -14,7 +14,8 @@ translator = Translator()
14
  # Get languages available for translation
15
  languages = []
16
  for code, name in googletrans.LANGUAGES.items():
17
- languages.append((code, name.capitalize()))
 
18
 
19
  def download_from_youtube(url):
20
  """
@@ -74,12 +75,12 @@ def divide_into_30s_segments(audio_fpath, seconds_max):
74
 
75
  return segment_paths, segment_start_times
76
 
77
- def get_translation(text, dest="en"):
78
  """
79
  Translates the given Swedish text to the language specified.
80
  """
81
- dest_text = dest[0]
82
- result = translator.translate(text, dest_text, 'sv')
83
  return result.text
84
 
85
 
 
14
  # Get languages available for translation
15
  languages = []
16
  for code, name in googletrans.LANGUAGES.items():
17
+ language = f"{name.capitalize()} ({code})"
18
+ languages.append(language)
19
 
20
  def download_from_youtube(url):
21
  """
 
75
 
76
  return segment_paths, segment_start_times
77
 
78
+ def get_translation(text, dest="English (en)"):
79
  """
80
  Translates the given Swedish text to the language specified.
81
  """
82
+ lang_code = dest.split(" ")[-1][1:-1]
83
+ result = translator.translate(text, lang_code, 'sv')
84
  return result.text
85
 
86
 
requirements.txt CHANGED
@@ -7,4 +7,4 @@ torch
7
  torchaudio
8
  moviepy
9
  git+https://github.com/pytube/pytube
10
- googletrans
 
7
  torchaudio
8
  moviepy
9
  git+https://github.com/pytube/pytube
10
+ googletrans==4.0.0rc1