Update README.md

README.md

@@ -103,6 +103,15 @@ Thank you.
 최근 인공지능에 관한 관심이 높아지면서 많은 사람이 인공지능 관련 수업 및 프로젝트를 진행하고 있습니다. 그러나 인공지능 관련 대학원생으로서 이러한 수요에 비해 인공지능 초보자들이 잘 알아들을 수 있는 유용한 자료는 흔치 않습니다. 더불어 각자의 수준과 분야에 개인화된 강의 또한 부족한 상황이어서 많은 사람들이 인공지능 학습을 시작하기 어려워하고 있습니다. 이러한 문제를 해결하고자, 저희 팀은 인공지능 용어 도메인에서 과외 선생님 역할을 하는 언어모델을 만들었습니다. 모델의 종류, 학습 데이터셋, 사용법 등이 아래에 설명되어 있으니 자세히 읽어보시고, 꼭 사용해 보시기 바랍니다.
 
 
+## How to use?
+
+
+<img src="https://github.com/CountingMstar/AI_BERT/assets/90711707/45afcd24-7ef9-4149-85d4-2236e23fbf69" width="1000" height="400"/>
+
+
+위 학습 과정을 통해 학습된 모델을 불러와 사용하시면 됩니다.
+
+
 ## Model
 https://huggingface.co/bert-base-uncased
 
@@ -173,17 +182,6 @@ optim = AdamW(model.parameters(), lr=5e-5)
 결과는 위 그래프들과 같이 마지막 에포크 기준 loss = 6.917126256477786, accuracy = 0.9819078947368421로 상당히 학습이 잘 된 모습을 보여줍니다.
 
 
-
-## How to use?
-
-
-```
-model = torch.load("./models/AI_BERT_final_10.pth")
-```
-
-위 학습 과정을 통해 학습된 모델을 불러와 사용하시면 됩니다.
-
-
 감사합니다.