Artificial-superintelligence commited on
Commit
dba57d9
·
verified ·
1 Parent(s): 2131dcb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -1,14 +1,14 @@
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,7 +51,8 @@ if video_file:
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
 
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