Update README.md

README.md

@@ -8,7 +8,7 @@ datasets:
 
 # Adapter `allenai/specter2_classification` for allenai/specter2
 
-An [adapter](https://adapterhub.ml) for the `allenai/specter2` model that was trained on the [allenai/scirepeval](https://huggingface.co/datasets/allenai/scirepeval/) dataset.
+An [adapter](https://adapterhub.ml) for the [`allenai/specter2`](https://huggingface.co/allenai/specter2_base) model that was trained on the [allenai/scirepeval](https://huggingface.co/datasets/allenai/scirepeval/) dataset.
 
 This adapter was created for usage with the **[adapter-transformers](https://github.com/Adapter-Hub/adapter-transformers)** library.
 
@@ -26,7 +26,7 @@ Now, the adapter can be loaded and activated like this:
 ```python
 from transformers import AutoAdapterModel
 
-model = AutoAdapterModel.from_pretrained("allenai/specter2")
+model = AutoAdapterModel.from_pretrained("allenai/specter2_base")
 adapter_name = model.load_adapter("allenai/specter2_classification", source="hf", set_active=True)
 ```
 
@@ -87,7 +87,7 @@ It builds on the work done in [SciRepEval: A Multi-Format Benchmark for Scientif
 
 |Model|Name and HF link|Description|
 |--|--|--|
-|Retrieval*|[allenai/specter2_proximity](https://huggingface.co/allenai/specter2_proximity)|Encode papers as queries and candidates eg. Link Prediction, Nearest Neighbor Search|
+|Retrieval*|[allenai/specter2_proximity](https://huggingface.co/allenai/specter2)|Encode papers as queries and candidates eg. Link Prediction, Nearest Neighbor Search|
 |Adhoc Query|[allenai/specter2_adhoc_query](https://huggingface.co/allenai/specter2_adhoc_query)|Encode short raw text queries for search tasks. (Candidate papers can be encoded with proximity)|
 |Classification|[allenai/specter2_classification](https://huggingface.co/allenai/specter2_classification)|Encode papers to feed into linear classifiers as features|
 |Regression|[allenai/specter2_regression](https://huggingface.co/allenai/specter2_regression)|Encode papers to feed into linear regressors as features|
@@ -98,10 +98,10 @@ It builds on the work done in [SciRepEval: A Multi-Format Benchmark for Scientif
 from transformers import AutoTokenizer, AutoModel
 
 # load model and tokenizer
-tokenizer = AutoTokenizer.from_pretrained('allenai/specter2')
+tokenizer = AutoTokenizer.from_pretrained('allenai/specter2_base')
 
 #load base model
-model = AutoModel.from_pretrained('allenai/specter2')
+model = AutoModel.from_pretrained('allenai/specter2_base')
 
 #load the adapter(s) as per the required task, provide an identifier for the adapter in load_as argument and activate it
 model.load_adapter("allenai/specter2_classification", source="hf", load_as="classification", set_active=True)