Update app.py
Browse files
app.py
CHANGED
@@ -4,7 +4,7 @@ from transformers import pipeline
|
|
4 |
|
5 |
auth_token = os.environ.get("access_token")
|
6 |
pipeline_en = pipeline(task="text-classification", model="Hello-SimpleAI/chatgpt-qa-detector-distil",use_auth_token=auth_token)
|
7 |
-
|
8 |
|
9 |
|
10 |
|
@@ -13,7 +13,7 @@ def predict_en(q,a):
|
|
13 |
return res['label'],res['score']
|
14 |
|
15 |
def predict_zh(q,a):
|
16 |
-
res =
|
17 |
return res['label'],res['score']
|
18 |
|
19 |
|
|
|
4 |
|
5 |
auth_token = os.environ.get("access_token")
|
6 |
pipeline_en = pipeline(task="text-classification", model="Hello-SimpleAI/chatgpt-qa-detector-distil",use_auth_token=auth_token)
|
7 |
+
pipeline_zh = pipeline(task="text-classification", model="Hello-SimpleAI/chatgpt-qa-detector-chinese",use_auth_token=auth_token)
|
8 |
|
9 |
|
10 |
|
|
|
13 |
return res['label'],res['score']
|
14 |
|
15 |
def predict_zh(q,a):
|
16 |
+
res = pipeline_zh({"text":q, "text_pair":a})
|
17 |
return res['label'],res['score']
|
18 |
|
19 |
|