Update app.py

app.py

@@ -61,11 +61,11 @@ transformer_model.eval()
 # llama_model = LlamaForCausalLM.from_pretrained(llama_model_name, device_map="auto")
 # llama_model.eval()
 
-from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
+from transformers import AutoTokenizer, AutoModelForCausalLM
 
-indic_model_name = "ai4bharat/indictrans2-en-hi"
+indic_model_name = "mistralai/Mistral-7B-Instruct-v0.3"
 indic_tokenizer = AutoTokenizer.from_pretrained(indic_model_name)
-indic_model = AutoModelForSeq2SeqLM.from_pretrained(indic_model_name)
+indic_model = AutoModelForCausalLM.from_pretrained(indic_model_name)
 
 # ----------------------
 # 4️⃣ Transliteration Function