Update README.md

README.md

@@ -17,6 +17,7 @@ Model trained from [mobileBert](https://huggingface.co/google/mobilebert-uncased
 <div align="center">
   <a href=https://github.com/04AR/Senti target="_blank"><img src=https://img.shields.io/badge/Code-black.svg?logo=github height=22px></a>
   <a href=https://huggingface.co/AR04/Senti target="_blank"><img src=https://img.shields.io/badge/%F0%9F%A4%97%20Models-d96902.svg height=22px></a>
+  <a href=https://github.com/04AR/Senti/blob/main/Senti.ipynb target="_blank"><img src="https://img.shields.io/badge/Training%20Code-%20jupyter-orange"></a>
 </div>
 
 #### Dataset used for the model
@@ -53,7 +54,7 @@ pip install torch transformers
 ```python
 from transformers import pipeline
 classifier = pipeline(task="text-classification", model="AR04/Senti", top_k=None)
-sentences = ["hi! u r looki beautiful today dear"]
+sentences = ["hi! u r looking beautiful today dear"]
 model_outputs = classifier(sentences)
 print(model_outputs[0])
 # produces a list of dicts for each of the labels