Update app.py

app.py

@@ -39,7 +39,7 @@ 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.get_lang_id(target_lang))
+    translated = model.generate(encoded_text, forced_bos_token_id=tokenizer.lang_code_to_id[target_lang])
     translated_text = tokenizer.decode(translated[0], skip_special_tokens=True)
     return translated_text
 
@@ -64,3 +64,4 @@ if st.button("Translate and Synthesize Speech"):
     else:
         st.audio(audio_bytes, format="audio/wav")
 
+