Update app.py

app.py

@@ -158,10 +158,13 @@ def process_video(video_file, language_choice):
                 start, end, translated_text = segment
                 segment_path = os.path.join(output_dir, f'segment_{start}_{end}.wav')
                 model.tts_to_file(translated_text, speaker_id, segment_path, speed=speed)
-                print(f"{start}, {end}")
+
                 reference_speaker = AudioFileClip.subclip(audio_clip, start, end)  # This is the voice you want to clone
                 reference_speaker.write_audiofile(f'reference_speaker_{start}_{end}.wav')
-                target_se, audio_name = se_extractor.get_se(f'reference_speaker_{start}_{end}.wav', tone_color_converter, vad=False)
+                try:
+                    target_se, audio_name = se_extractor.get_se(f'reference_speaker_{start}_{end}.wav', tone_color_converter, vad=False)
+                except NotImplementedError:
+                    target_se, audio_name = se_extractor.get_se(reference_audio, tone_color_converter, vad=False)
                 # Run the tone color converter
                 encode_message = "@MyShell"
                 tone_color_converter.convert(