qqwjq1981 commited on
Commit
0ab292a
·
verified ·
1 Parent(s): 15d6241

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -153,9 +153,9 @@ def get_translation_model(source_language, target_language):
153
  # Return the model using string concatenation
154
  return f"Helsinki-NLP/opus-mt-{source_language}-{target_language}"
155
 
156
- def translate_single_entry(entry, translator, target_language):
157
  original_text = entry["text"]
158
- translated_text = translator(original_text, target_language=target_language)[0]['translation_text']
159
  return {
160
  "start": entry["start"],
161
  "original": original_text,
@@ -172,7 +172,7 @@ def translate_text(transcription_json, source_language, target_language):
172
  # Use ThreadPoolExecutor to parallelize translations
173
  with concurrent.futures.ThreadPoolExecutor() as executor:
174
  # Submit all translation tasks and collect results
175
- translate_func = lambda entry: translate_single_entry(entry, translator, target_language)
176
  translated_json = list(executor.map(translate_func, transcription_json))
177
 
178
  # Sort the translated_json by start time
 
153
  # Return the model using string concatenation
154
  return f"Helsinki-NLP/opus-mt-{source_language}-{target_language}"
155
 
156
+ def translate_single_entry(entry, translator):
157
  original_text = entry["text"]
158
+ translated_text = translator(original_text)[0]['translation_text']
159
  return {
160
  "start": entry["start"],
161
  "original": original_text,
 
172
  # Use ThreadPoolExecutor to parallelize translations
173
  with concurrent.futures.ThreadPoolExecutor() as executor:
174
  # Submit all translation tasks and collect results
175
+ translate_func = lambda entry: translate_single_entry(entry, translator)
176
  translated_json = list(executor.map(translate_func, transcription_json))
177
 
178
  # Sort the translated_json by start time