Update README.md

README.md

@@ -36,7 +36,7 @@ tokenizer = AutoTokenizer.from_pretrained(model_path)
 
 def translate(
     text: str,
-    to: Literal["chinese", "english"] = "chinese",
+    to: Literal["Chinese", "English"] = "Chinese",
     **kwargs
 ):
     generation_args = dict(
@@ -73,7 +73,7 @@ def translate(
     return response
 
 text = "After a long day at work, I love to unwind by cooking a nice dinner and watching my favorite TV series. It really helps me relax and recharge for the next day."
-response = translate(text=text, to='chinese')
+response = translate(text=text, to='Chinese')
 print(f'Translation: {response}')
 
 """