rasyosef commited on
Commit
f6ef66a
·
verified ·
1 Parent(s): 011f837

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -5
README.md CHANGED
@@ -37,15 +37,12 @@ from sentence_transformers import SparseEncoder
37
  from splade_index import SPLADE
38
 
39
  # Download the SPLADE model that was used to create the index from the HuggingFace Hub
40
- model_id = "the-splade-model-id" # Enter the splade model id
41
  model = SparseEncoder(model_id)
42
 
43
- # Set your huggingface token if repo is private
44
- token = os.environ["HF_TOKEN"]
45
  repo_id = "rasyosef/msmarco_dev_1M_splade_index"
46
-
47
  # Load a SPLADE index from the Hugging Face model hub
48
- retriever = SPLADE.load_from_hub(repo_id, model=model, token=token)
49
  ```
50
 
51
  ## Stats
 
37
  from splade_index import SPLADE
38
 
39
  # Download the SPLADE model that was used to create the index from the HuggingFace Hub
40
+ model_id = "rasyosef/splade-mini" # The SPLADE model id
41
  model = SparseEncoder(model_id)
42
 
 
 
43
  repo_id = "rasyosef/msmarco_dev_1M_splade_index"
 
44
  # Load a SPLADE index from the Hugging Face model hub
45
+ retriever = SPLADE.load_from_hub(repo_id, model=model)
46
  ```
47
 
48
  ## Stats