Spaces:

cn91
/

zaoju-demo

Sleeping

cn91 commited on Aug 16, 2023

Commit

9a919c2

•

1 Parent(s): b363777

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -11,8 +11,8 @@ if USE_GPU and torch.cuda.is_available():
 else:
     device = torch.device('cpu')
-MODEL_NAME_CHINESE = "IDEA-CCNL/Erlangshen-DeBERTa-v2-186M-Chinese-SentencePiece"
-#MODEL_NAME_CHINESE = "IDEA-CCNL/Erlangshen-DeBERTa-v2-97M-CWS-Chinese"
 WORD_PROBABILITY_THRESHOLD = 0.02

 else:
     device = torch.device('cpu')
+#MODEL_NAME_CHINESE = "IDEA-CCNL/Erlangshen-DeBERTa-v2-186M-Chinese-SentencePiece"
+MODEL_NAME_CHINESE = "IDEA-CCNL/Erlangshen-DeBERTa-v2-97M-CWS-Chinese"
 WORD_PROBABILITY_THRESHOLD = 0.02