Spaces:
Build error
Build error
Update CrossEncoder/cross_encoder.py
Browse files
CrossEncoder/cross_encoder.py
CHANGED
|
@@ -92,7 +92,7 @@ class CERank:
|
|
| 92 |
def get_ranked_contexts(context_json, question):
|
| 93 |
|
| 94 |
dirname = 'examples'
|
| 95 |
-
model_path = '
|
| 96 |
max_length = 512
|
| 97 |
|
| 98 |
# Can't use use_fast (fast tokenizers) while gradio is running, causes conflict with tokenizer multiprocessing/parallelism.
|
|
@@ -110,7 +110,7 @@ def get_ranked_contexts(context_json, question):
|
|
| 110 |
def get_ranked_contexts_in_memory(contexts, question):
|
| 111 |
|
| 112 |
dirname = 'examples'
|
| 113 |
-
model_path = '
|
| 114 |
max_length = 512
|
| 115 |
|
| 116 |
# Can't use use_fast (fast tokenizers) while gradio is running, causes conflict with tokenizer multiprocessing/parallelism.
|
|
|
|
| 92 |
def get_ranked_contexts(context_json, question):
|
| 93 |
|
| 94 |
dirname = 'examples'
|
| 95 |
+
model_path = 'ms-marco-electra-base'
|
| 96 |
max_length = 512
|
| 97 |
|
| 98 |
# Can't use use_fast (fast tokenizers) while gradio is running, causes conflict with tokenizer multiprocessing/parallelism.
|
|
|
|
| 110 |
def get_ranked_contexts_in_memory(contexts, question):
|
| 111 |
|
| 112 |
dirname = 'examples'
|
| 113 |
+
model_path = 'ms-marco-electra-base'
|
| 114 |
max_length = 512
|
| 115 |
|
| 116 |
# Can't use use_fast (fast tokenizers) while gradio is running, causes conflict with tokenizer multiprocessing/parallelism.
|