Update README.md

README.md

@@ -1,7 +1,9 @@
 ---
-license: cc-by-3.0
 language:
 - en
+tags:
+- Sentence Similarity
+- feature-extraction
 ---
 
 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.
@@ -14,9 +16,9 @@ import torch
 def load_finetuned_model():
 
 
-        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")
+        sentence_encoder = AutoModel.from_pretrained("biu-nlp/abstract-sim-sentence")
+        query_encoder = AutoModel.from_pretrained("biu-nlp/abstract-sim-query")
+        tokenizer = AutoTokenizer.from_pretrained("biu-nlp/abstract-sim-sentence")
 
         return tokenizer, query_encoder, sentence_encoder