Update app.py

app.py

@@ -22,13 +22,14 @@ def main():
 
 def translate_text(input_text, model, tokenizer):
     # Tokenize input text
-    input_ids = tokenizer(input_text, return_tensors="pt").input_ids
-
-    # Generate translation
-    translated_ids = model.generate(input_ids)
+    model_inputs = tokenizer(input_text, return_tensors="pt").input_ids
+    generated_tokens = model.generate(
+    **model_inputs,
+    forced_bos_token_id=tokenizer.lang_code_to_id["hi_IN"]
+)
 
     # Decode translated text
-    translated_text = tokenizer.decode(translated_ids[0], skip_special_tokens=True)
+    translated_text = tokenizer.decode(generated_tokens[0], skip_special_tokens=True)
 
     return translated_text