[update]add main

main.py

@@ -36,7 +36,9 @@ def main():
                                ):
         tokenizer.src_lang = src_lang
         encoded_src = tokenizer(src_text, return_tensors="pt")
-        generated_tokens = model.generate(**encoded_src, forced_bos_token_id=tokenizer.get_lang_id(tgt_lang))
+        generated_tokens = model.generate(**encoded_src,
+                                          forced_bos_token_id=tokenizer.get_lang_id(tgt_lang),
+                                          )
         result = tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)
 
         return result