Spaces:
Running
Running
Update app.py
Browse files
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-
|
11 |
-
model = AutoModelForMaskedLM.from_pretrained("naver/splade-
|
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:
|