Update app.py
Browse files
app.py
CHANGED
@@ -1,14 +1,14 @@
|
|
1 |
import streamlit as st
|
2 |
from moviepy.editor import VideoFileClip
|
3 |
import whisper
|
4 |
-
from
|
5 |
from gtts import gTTS
|
6 |
import tempfile
|
7 |
import os
|
8 |
|
9 |
# Initialize Whisper model and translator
|
10 |
whisper_model = whisper.load_model("base") # Choose model size: 'tiny', 'base', 'small', 'medium', 'large'
|
11 |
-
translator = Translator()
|
12 |
|
13 |
# Language options
|
14 |
LANGUAGES = {
|
@@ -51,7 +51,8 @@ if video_file:
|
|
51 |
st.write("Original Transcription:", original_text)
|
52 |
|
53 |
# Translate text to the target language
|
54 |
-
|
|
|
55 |
st.write(f"Translated Text ({target_language}):", translated_text)
|
56 |
|
57 |
# Convert translated text to speech
|
|
|
1 |
import streamlit as st
|
2 |
from moviepy.editor import VideoFileClip
|
3 |
import whisper
|
4 |
+
from translate import Translator # Updated import
|
5 |
from gtts import gTTS
|
6 |
import tempfile
|
7 |
import os
|
8 |
|
9 |
# Initialize Whisper model and translator
|
10 |
whisper_model = whisper.load_model("base") # Choose model size: 'tiny', 'base', 'small', 'medium', 'large'
|
11 |
+
translator = Translator(to_lang="en") # Updated to specify target language dynamically
|
12 |
|
13 |
# Language options
|
14 |
LANGUAGES = {
|
|
|
51 |
st.write("Original Transcription:", original_text)
|
52 |
|
53 |
# Translate text to the target language
|
54 |
+
translator = Translator(to_lang=LANGUAGES[target_language]) # Dynamically set language
|
55 |
+
translated_text = translator.translate(original_text)
|
56 |
st.write(f"Translated Text ({target_language}):", translated_text)
|
57 |
|
58 |
# Convert translated text to speech
|