Update README.md

README.md

@@ -40,13 +40,11 @@ tobe_translated = ['講筵官','判司簿尉','散騎常侍','殿中省尚輦奉
 
 <font color="cornflowerblue"> 3. Make a prediction </font>
 ```python
-trans_list = []
-for chinese_sentence in tobe_translated:
-    inputs = tokenizer([chinese_sentence], return_tensors="pt", padding=True).to(device)
-    translated = model.generate(**inputs, max_length=128)
-    tran = [tokenizer.decode(t, skip_special_tokens=True) for t in translated][0]
-    print(f'{chinese_sentence}: {tran}')
-    trans_list.append(tran[0])
+inputs = tokenizer(tobe_translated, return_tensors="pt", padding=True).to(device)
+translated = model.generate(**inputs, max_length=128)
+tran = [tokenizer.decode(t, skip_special_tokens=True) for t in translated]
+for c, t in zip(tobe_translated, tran):
+    print(f'{c}: {t}')
 ```
 講筵官: Lecturer<br>
 判司簿尉: Supervisor of the Commandant of Records<br>