Update main.py

main.py

@@ -33,11 +33,13 @@ class TranslationRequest(BaseModel):
     src_lang: str
     tgt_lang: str
 
+
 @app.post("/translate")
 def translate(req: TranslationRequest):
     inputs = tokenizer(
         req.text,
         return_tensors="pt",
+        src_lang=req.src_lang   # ✅ set source language
     ).to(model.device)
 
     tgt_lang = req.tgt_lang   # e.g. "bam_Latn"