Update app.py
Browse files
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
|
157 |
original_text = entry["text"]
|
158 |
-
translated_text = translator(original_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
|
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
|