Update app.py

app.py

@@ -1,17 +1,7 @@
 import streamlit as st
 from transformers import MarianMTModel, MarianTokenizer, pipeline
-import torch
 from io import BytesIO
 
-# Define the translation model
-model_name = "Helsinki-NLP/opus-mt-en-es"  # Default to Spanish for the example
-tokenizer = MarianTokenizer.from_pretrained(model_name)
-model = MarianMTModel.from_pretrained(model_name)
-
-# Define the TTS model
-tts_model_name = "microsoft/speecht5_tts"
-tts_pipeline = pipeline("text-to-speech", model=tts_model_name)
-
 # Define the language options
 languages = {
     'es': 'Spanish',
@@ -39,14 +29,16 @@ def translate_text(text, target_lang):
     
     # Translate text
     encoded_text = tokenizer.encode(text, return_tensors="pt")
-    translated = model.generate(encoded_text, forced_bos_token_id=tokenizer.lang_code_to_id[target_lang])
+    forced_bos_token_id = model.config.forced_bos_token_id
+    translated = model.generate(encoded_text, forced_bos_token_id=forced_bos_token_id)
     translated_text = tokenizer.decode(translated[0], skip_special_tokens=True)
     return translated_text
 
 def synthesize_speech(text, lang):
     # Synthesize speech
+    tts_pipeline = pipeline("text-to-speech", model="microsoft/speecht5_tts")
     try:
-        audio = tts_pipeline(text, lang=lang)
+        audio = tts_pipeline(text)
         audio_bytes = BytesIO(audio["audio"])
         return audio_bytes, None
     except Exception as e:
@@ -65,3 +57,4 @@ if st.button("Translate and Synthesize Speech"):
         st.audio(audio_bytes, format="audio/wav")
 
 
+