Raghav001 commited on
Commit
33106aa
Β·
1 Parent(s): ab98b8b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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('uer/sbert-base-chinese-nli', 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()
 
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()