Update app.py
Browse files
app.py
CHANGED
|
@@ -7,7 +7,7 @@ import pandas as pd
|
|
| 7 |
import time
|
| 8 |
from cnocr import CnOcr
|
| 9 |
from sentence_transformers import SentenceTransformer, models, util
|
| 10 |
-
word_embedding_model = models.Transformer('
|
| 11 |
pooling_model = models.Pooling(word_embedding_model.get_word_embedding_dimension(), pooling_mode='cls')
|
| 12 |
embedder = SentenceTransformer(modules=[word_embedding_model, pooling_model])
|
| 13 |
ocr = CnOcr()
|
|
|
|
| 7 |
import time
|
| 8 |
from cnocr import CnOcr
|
| 9 |
from sentence_transformers import SentenceTransformer, models, util
|
| 10 |
+
word_embedding_model = models.Transformer('sentence-transformers/all-MiniLM-L6-v2', do_lower_case=True)
|
| 11 |
pooling_model = models.Pooling(word_embedding_model.get_word_embedding_dimension(), pooling_mode='cls')
|
| 12 |
embedder = SentenceTransformer(modules=[word_embedding_model, pooling_model])
|
| 13 |
ocr = CnOcr()
|