Spaces:
Sleeping
Sleeping
Update Milestone5API.py
Browse files- Milestone5API.py +2 -0
Milestone5API.py
CHANGED
|
@@ -35,8 +35,10 @@ def text_to_speech(translated_text_file_path, video_path):
|
|
| 35 |
|
| 36 |
output_video_path = f"CS370_Milestone5/videos/{file_name_only}_new.mp4"
|
| 37 |
|
|
|
|
| 38 |
replace_video_audio(video_path, output_file_path, output_video_path)
|
| 39 |
return output_video_path
|
|
|
|
| 40 |
def translate_text_file(input_text_path, output_text_path, source_lang='en', target_lang='fr', model_name="Helsinki-NLP/opus-mt-en-fr", batch_size=8):
|
| 41 |
model = MarianMTModel.from_pretrained(model_name)
|
| 42 |
tokenizer = MarianTokenizer.from_pretrained(model_name)
|
|
|
|
| 35 |
|
| 36 |
output_video_path = f"CS370_Milestone5/videos/{file_name_only}_new.mp4"
|
| 37 |
|
| 38 |
+
# Replace the video's audio with the generated audio
|
| 39 |
replace_video_audio(video_path, output_file_path, output_video_path)
|
| 40 |
return output_video_path
|
| 41 |
+
|
| 42 |
def translate_text_file(input_text_path, output_text_path, source_lang='en', target_lang='fr', model_name="Helsinki-NLP/opus-mt-en-fr", batch_size=8):
|
| 43 |
model = MarianMTModel.from_pretrained(model_name)
|
| 44 |
tokenizer = MarianTokenizer.from_pretrained(model_name)
|