Update app.py
Browse files
app.py
CHANGED
@@ -1,13 +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
|
10 |
whisper_model = whisper.load_model("base") # Choose model size: 'tiny', 'base', 'small', 'medium', 'large'
|
|
|
11 |
|
12 |
# Language options
|
13 |
LANGUAGES = {
|
@@ -50,7 +51,7 @@ if video_file:
|
|
50 |
st.write("Original Transcription:", original_text)
|
51 |
|
52 |
# Translate text to the target language
|
53 |
-
translated_text =
|
54 |
st.write(f"Translated Text ({target_language}):", translated_text)
|
55 |
|
56 |
# Convert translated text to speech
|
|
|
1 |
import streamlit as st
|
2 |
from moviepy.editor import VideoFileClip
|
3 |
import whisper
|
4 |
+
from googletrans import Translator
|
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 |
st.write("Original Transcription:", original_text)
|
52 |
|
53 |
# Translate text to the target language
|
54 |
+
translated_text = translator.translate(original_text, dest=LANGUAGES[target_language]).text
|
55 |
st.write(f"Translated Text ({target_language}):", translated_text)
|
56 |
|
57 |
# Convert translated text to speech
|