Spaces:
Running
Running
File size: 252 Bytes
4c2a969 |
1 2 3 4 5 6 7 8 9 |
INDEX_DIR = 'data/index'
STATEMENTS_PATH = 'data/statements.txt'
RETRIEVER_MODEL = "sentence-transformers/msmarco-distilbert-base-tas-b"
RETRIEVER_MODEL_FORMAT = "sentence_transformers"
RETRIEVER_TOP_K = 5
NLI_MODEL = "valhalla/distilbart-mnli-12-1"
|