Update app.py

app.py

@@ -79,8 +79,8 @@ def batch_translate(input_sentences, src_lang, tgt_lang, model, tokenizer, ip):
     return translations
 
 # Initialize IndicTrans2
-en_indic_ckpt_dir = "ai4bharat/indictrans2-indic-en-1B"
-en_indic_tokenizer, en_indic_model = initialize_model_and_tokenizer(en_indic_ckpt_dir, quantization)
+indic_en_ckpt_dir = "ai4bharat/indictrans2-indic-en-1B"
+indic_en_tokenizer, indic_en_model = initialize_model_and_tokenizer(indic_en_ckpt_dir, quantization)
 ip = IndicProcessor(inference=True)
 
 # ---- Gradio Function ----
@@ -99,8 +99,8 @@ def transcribe_and_translate(audio):
     # Translation
     en_sents = [malayalam_text]
     src_lang, tgt_lang = "mal_Mlym", "eng_Latn"
-    translations = batch_translate(en_sents, src_lang, tgt_lang, en_indic_model, en_indic_tokenizer, ip)
-    
+    translations = batch_translate(en_sents, src_lang, tgt_lang, indic_en_model, indic_en_tokenizer, ip)
+
     return malayalam_text, translations[0]
 
 # ---- Gradio Interface ----