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
|