JaMugen commited on
Commit
7ec85cd
·
1 Parent(s): 098d4f1

Update Milestone5API.py

Browse files
Files changed (1) hide show
  1. Milestone5API.py +1 -1
Milestone5API.py CHANGED
@@ -62,7 +62,7 @@ def text_to_speech(translated_text_file_path, video_path):
62
  new_video_path = f"CS370_Milestone5/videos/{file_name_only}_new2.mp4"
63
 
64
  replace_video_audio(video_path, output_file_path, output_video_path,translated_text,new_video_path)
65
- return new_video_path
66
 
67
  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):
68
  model = MarianMTModel.from_pretrained(model_name)
 
62
  new_video_path = f"CS370_Milestone5/videos/{file_name_only}_new2.mp4"
63
 
64
  replace_video_audio(video_path, output_file_path, output_video_path,translated_text,new_video_path)
65
+ return output_video_path
66
 
67
  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):
68
  model = MarianMTModel.from_pretrained(model_name)