Update README.md

README.md

@@ -1,3 +1,7 @@
+# A Multi-task learning model with two prediction heads
+* First prediction head classifies between keyword sentences vs statements/questions
+* Second prediction head corresponds to classifier for statements vs questions
+
 ## Article
 [Medium article](https://medium.com/@shahrukhx01/multi-task-learning-with-transformers-part-1-multi-prediction-heads-b7001cf014bf)
 ## Demo Notebook