Update README.md
Browse files
README.md
CHANGED
@@ -43,7 +43,7 @@ Then you can use the model like this:
|
|
43 |
```python
|
44 |
from sentence_transformers import CrossEncoder
|
45 |
|
46 |
-
reranker_model = CrossEncoder('DiTy/cross-encoder-russian-msmarco', device='cuda')
|
47 |
|
48 |
query = ["как часто нужно ходить к стоматологу?"]
|
49 |
documents = [
|
|
|
43 |
```python
|
44 |
from sentence_transformers import CrossEncoder
|
45 |
|
46 |
+
reranker_model = CrossEncoder('DiTy/cross-encoder-russian-msmarco', max_length=512, device='cuda')
|
47 |
|
48 |
query = ["как часто нужно ходить к стоматологу?"]
|
49 |
documents = [
|