Spaces:

cn91
/

zaoju-demo

Sleeping

cn91 commited on Aug 16, 2023

Commit

b363777

•

1 Parent(s): 02416b1

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