cn91 commited on
Commit
b6df2fe
1 Parent(s): 2609434

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -0
app.py CHANGED
@@ -11,6 +11,12 @@ if USE_GPU and torch.cuda.is_available():
11
  else:
12
  device = torch.device('cpu')
13
 
 
 
 
 
 
 
14
 
15
  WORD_PROBABILITY_THRESHOLD = 0.02
16
  #WORD_PROBABILITY_THRESHOLD_ENGLISH = 0.02
 
11
  else:
12
  device = torch.device('cpu')
13
 
14
+ MODEL_NAME_ENGLISH = "facebook/xlm-v-base"
15
+ #SENTENCE_MODEL_NAME_ENGLISH = 'sentence-transformers/all-MiniLM-L6-v2'
16
+ #WORD_MODEL_NAME_ENGLISH = 'vocab-transformers/distilbert-word2vec_256k-MLM_best'
17
+
18
+ # chinese models
19
+ MODEL_NAME_CHINESE = "IDEA-CCNL/Erlangshen-DeBERTa-v2-186M-Chinese-SentencePiece"
20
 
21
  WORD_PROBABILITY_THRESHOLD = 0.02
22
  #WORD_PROBABILITY_THRESHOLD_ENGLISH = 0.02