Update README.md

README.md

@@ -126,6 +126,13 @@ predict_score = extract_score(output_text)
 print(predict_score) # 1
 ```
 
+### **Heatmap Visualize**
+We randomly sampled 200 sentences from the training data, extracted scores from the model-generated sentences, and compared them to the correct answers.
+The training and test datasets are not separated, so we can only see how well the model learned.
+- prometheus-7b-v1.0 (english train-> english inference) # 3 failed to output a score, total 197
+- llama3-8b-it-prometheus-ko (korean train-> korean inference) # total 200
+![image/png](https://cdn-uploads.huggingface.co/production/uploads/6152b4b9ecf3ca6ab820e325/ssZRGTysyiOZD4ttNOD4s.png)
+
 ### **Citation**
 ```bibtex
 @misc{kim2023prometheus,
@@ -137,4 +144,4 @@ print(predict_score) # 1
     primaryClass={cs.CL}
 }
 ```
-Our trainig code can be found here: [TBD]
+Our trainig code can be found here: [TBD]