Upload app.py

app.py

@@ -11,12 +11,12 @@ pipeline_zh = pipeline(task="text-classification", model="MeiJuice/CheckGPT-Chin
 
 def predict_en(text):
     res = pipeline_en(text)[0]
-    return res['label'], res['score']
+    return "ChatGPT" if res['label'] == "LABEL_1" else "human", res['score']
 
 
 def predict_zh(text):
     res = pipeline_zh(text)[0]
-    return res['label'], res['score']
+    return "ChatGPT" if res['label'] == "LABEL_1" else "human", res['score']
 
 
 with gr.Blocks() as demo:
@@ -25,8 +25,7 @@ with gr.Blocks() as demo:
                     Note: Providing more text to the `Text` box can make the prediction more accurate!
                     """)
         t1 = gr.Textbox(lines=5, label='Text',
-                        value="I worked on a new function on my child theme which consists of displaying on the product the number of times it has already been added to the cart. Example: if I have 3 tshirts and 1 jeans in my basket, it must respectively appear under the article tshirt > 3 and under the article jeans > 1\
-I have already managed to make them appear individually. Unfortunately, they all appear after AND under all products.")
+                        value="There are a few things that can help protect your credit card information from being misused when you give it to a restaurant or any other business:\n\nEncryption: Many businesses use encryption to protect your credit card information when it is being transmitted or stored. This means that the information is transformed into a code that is difficult for anyone to read without the right key.")
         button1 = gr.Button("🤖 Predict!")
         label1 = gr.Textbox(lines=1, label='Predicted Label 🎃')
         score1 = gr.Textbox(lines=1, label='Prob')
@@ -35,7 +34,7 @@ I have already managed to make them appear individually. Unfortunately, they all
                     注意: 在`文本`栏中输入更多的文本，可以让预测更准确哦！
                     """)
         t2 = gr.Textbox(lines=5, label='文本',
-                        value="整体呈现 U 字形。左边叫做收缩路径（contracting path），主要是max pool 下采样，同时通道数增加。右边是扩张路径（expansive path），用反卷积（ up-convolution ）做上采样，恢复图像的尺寸。最后采用1\times1的卷积层代替全连接层输出。")
+                        value="对于OpenAI大力出奇迹的工作，自然每个人都有自己的看点。我自己最欣赏的地方是ChatGPT如何解决 “AI校正(Alignment)“这个问题。这个问题也是我们课题组这两年在探索的学术问题之一。")
         button2 = gr.Button("🤖 预测!")
         label2 = gr.Textbox(lines=1, label='预测结果 🎃')
         score2 = gr.Textbox(lines=1, label='模型概率')