Update app.py

app.py

@@ -1,14 +1,16 @@
 import streamlit as st
 from moviepy.editor import VideoFileClip
 import whisper
-from translate import Translator  # Updated import
+from translate import Translator
 from gtts import gTTS
 import tempfile
 import os
 
-# Initialize Whisper model and translator
-whisper_model = whisper.load_model("base")  # Choose model size: 'tiny', 'base', 'small', 'medium', 'large'
-translator = Translator(to_lang="en")  # Updated to specify target language dynamically
+# Initialize Whisper model
+try:
+    whisper_model = whisper.load_model("base")  # Ensure the model is installed from the correct Whisper library
+except AttributeError:
+    st.error("Whisper model could not be loaded. Ensure that Whisper is installed from GitHub.")
 
 # Language options
 LANGUAGES = {
@@ -51,7 +53,7 @@ if video_file:
             st.write("Original Transcription:", original_text)
 
             # Translate text to the target language
-            translator = Translator(to_lang=LANGUAGES[target_language])  # Dynamically set language
+            translator = Translator(to_lang=LANGUAGES[target_language])
             translated_text = translator.translate(original_text)
             st.write(f"Translated Text ({target_language}):", translated_text)