update README section 3

README.md

@@ -84,13 +84,17 @@ idx2label = {v:k for k,v in label2idx.items()}
 
 3. Change your sentence here
 ```python
+label2idx = {'not-letter': 0,'letter': 1}
+idx2label = {v:k for k,v in label2idx.items()}
+
 test_sen = '上丞相康思公書'
 pred_class_proba = predict_class(test_sen)
 print(f'The predicted probability for the {list(pred_class_proba.keys())[0]} class: {list(pred_class_proba.values())[0]}')
 print(f'The predicted probability for the {list(pred_class_proba.keys())[1]} class: {list(pred_class_proba.values())[1]}')
-
+>>> The predicted probability for the not-letter class: 0.002029061783105135
+>>> The predicted probability for the letter class: 0.9979709386825562
 pred_class = idx2label[np.argmax(list(pred_class_proba.values()))]
-print(f'The predicted class is: {pred_class}')test_sen = '上丞相康思公書'
-pred_class_dict = predict_class(test_sen)
-print(pred_class_dict)
+
+print(f'The predicted class is: {pred_class}')
+>>> The predicted class is: letter
 ```