Update README.md

README.md

@@ -4,7 +4,7 @@ language:
 - en
 ---
 
-A model for mapping abstract sentence descriptions to sentences that fit the descriptions. Use ```load_finetuned_model``` to load the query and sentence encoder, and ```encode_batch()``` to encode a sentence with the model.
+A model for mapping abstract sentence descriptions to sentences that fit the descriptions. Trained on Wikipedia. Use ```load_finetuned_model``` to load the query and sentence encoder, and ```encode_batch()``` to encode a sentence with the model.
 
 ```python
 
@@ -14,9 +14,9 @@ import torch
 def load_finetuned_model():
 
 
-        sentence_encoder = AutoModel.from_pretrained("ravfogs/abstract-sim-wiki-sentence")
-        query_encoder = AutoModel.from_pretrained("ravfogs/abstract-sim-wiki-query")
-        tokenizer = AutoTokenizer.from_pretrained("ravfogs/abstract-sim-wiki-sentence")
+        sentence_encoder = AutoModel.from_pretrained("ravfogs/abstract-sim-sentence")
+        query_encoder = AutoModel.from_pretrained("ravfogs/abstract-sim-query")
+        tokenizer = AutoTokenizer.from_pretrained("ravfogs/abstract-sim-sentence")
 
         return tokenizer, query_encoder, sentence_encoder