JaMugen commited on
Commit
5bdd7ea
·
1 Parent(s): a41d1df

Update Milestone5API.py

Browse files
Files changed (1) hide show
  1. 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)