SiddharthAK commited on
Commit
0ad0578
·
verified ·
1 Parent(s): 22b5137

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -7,8 +7,8 @@ import torch
7
  # "naver/splade-v3" is a common and robust SPLADE v3 model.
8
  # Make sure you've accepted any user access agreements on its Hugging Face Hub page.
9
  try:
10
- tokenizer = AutoTokenizer.from_pretrained("naver/splade-v3")
11
- model = AutoModelForMaskedLM.from_pretrained("naver/splade-v3")
12
  model.eval() # Set the model to evaluation mode for inference
13
  print("SPLADE v23 model and tokenizer loaded successfully!")
14
  except Exception as e:
 
7
  # "naver/splade-v3" is a common and robust SPLADE v3 model.
8
  # Make sure you've accepted any user access agreements on its Hugging Face Hub page.
9
  try:
10
+ tokenizer = AutoTokenizer.from_pretrained("naver/splade-cocondenser-selfdistil")
11
+ model = AutoModelForMaskedLM.from_pretrained("naver/splade-cocondenser-selfdistil")
12
  model.eval() # Set the model to evaluation mode for inference
13
  print("SPLADE v23 model and tokenizer loaded successfully!")
14
  except Exception as e: