Spaces:
Sleeping
Sleeping
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:
|