Update README.md

README.md

@@ -4,5 +4,11 @@ datasets:
 - sms1097/self_rag_tokens_train_data
 ---
 
-# Retrieval Prediction Model
-Classification model for an instruction model to determine if it requires additional context to answer a question. 
+# Retrieval Model
+
+This generates the `Retrieve` token as descirbed in Self-RAG. 
+
+We are testing to see if a retrieved document is relevant to the user input of our language model. 
+
+The expected input to the model is just a query posed to the language model. 
+