Artificial-superintelligence commited on
Commit
8472e1f
·
verified ·
1 Parent(s): 46617ca

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -1,13 +1,14 @@
1
  import streamlit as st
2
  from moviepy.editor import VideoFileClip
3
  import whisper
4
- from deep_translator import GoogleTranslator
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 = GoogleTranslator(source="auto", target=LANGUAGES[target_language]).translate(original_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