Update app.py

change model to indic languages

app.py

@@ -17,10 +17,10 @@ app.add_middleware(
 )
 
 model = AutoModelForSeq2SeqLM.from_pretrained(
-    "ai4bharat/indictrans2-en-indic-1B", trust_remote_code=True
+    "ai4bharat/indictrans2-indic-indic-1B", trust_remote_code=True
 )
 tokenizer = AutoTokenizer.from_pretrained(
-    "ai4bharat/indictrans2-en-indic-1B", trust_remote_code=True
+    "ai4bharat/indictrans2-indic-indic-1B", trust_remote_code=True
 )
 
 ip = IndicProcessor(inference=True)
@@ -28,9 +28,8 @@ DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 model = model.to(DEVICE)
 
 
-def translate_text(sentences: List[str], target_lang: str):
+def translate_text(sentences: List[str], src_lang: str, target_lang: str):
     try:
-        src_lang = "eng_Latn"
         batch = ip.preprocess_batch(sentences, src_lang=src_lang, tgt_lang=target_lang)
         inputs = tokenizer(
             batch,
@@ -68,6 +67,7 @@ def read_root():
 
 class TranslateRequest(BaseModel):
     sentences: List[str]
+    src_lang
     target_lang: str