Remove spacy download step; use pip-installed model
Browse files- categories/accuracy.py +1 -1
categories/accuracy.py
CHANGED
|
@@ -10,7 +10,7 @@ from transformers import AutoModel, AutoTokenizer
|
|
| 10 |
# setup global variables on import (bad practice, but whatever)
|
| 11 |
# --------------------------------------------------------------
|
| 12 |
|
| 13 |
-
aligner = SentenceAligner(model="
|
| 14 |
|
| 15 |
de_encoder = LaserEncoderPipeline(lang="deu_Latn")
|
| 16 |
en_encoder = LaserEncoderPipeline(lang="eng_Latn")
|
|
|
|
| 10 |
# setup global variables on import (bad practice, but whatever)
|
| 11 |
# --------------------------------------------------------------
|
| 12 |
|
| 13 |
+
aligner = SentenceAligner(model="bert-base-multilingual-cased", layer=6)
|
| 14 |
|
| 15 |
de_encoder = LaserEncoderPipeline(lang="deu_Latn")
|
| 16 |
en_encoder = LaserEncoderPipeline(lang="eng_Latn")
|