Upload app.py

app.py

@@ -11,6 +11,12 @@ if USE_GPU and torch.cuda.is_available():
 else: 
     device = torch.device('cpu')
 
+MODEL_NAME_ENGLISH = "facebook/xlm-v-base"
+#SENTENCE_MODEL_NAME_ENGLISH = 'sentence-transformers/all-MiniLM-L6-v2'
+#WORD_MODEL_NAME_ENGLISH = 'vocab-transformers/distilbert-word2vec_256k-MLM_best'
+
+# chinese models
+MODEL_NAME_CHINESE = "IDEA-CCNL/Erlangshen-DeBERTa-v2-186M-Chinese-SentencePiece"
 
 WORD_PROBABILITY_THRESHOLD = 0.02
 #WORD_PROBABILITY_THRESHOLD_ENGLISH = 0.02