beyond commited on
Commit
6cef871
1 Parent(s): a1df271

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +13 -29
app.py CHANGED
@@ -2,9 +2,9 @@ import os
2
  import gradio as gr
3
  from transformers import pipeline
4
 
5
- auth_token = os.environ.get("access_token")
6
- pipeline_en = pipeline(task="text-classification", model="Hello-SimpleAI/chatgpt-detector-single",use_auth_token=auth_token)
7
- pipeline_zh = pipeline(task="text-classification", model="Hello-SimpleAI/chatgpt-detector-single-chinese",use_auth_token=auth_token)
8
 
9
 
10
 
@@ -21,28 +21,9 @@ def predict_zh(text):
21
 
22
  with gr.Blocks() as demo:
23
  gr.Markdown("""
24
- ## ChatGPT Detector 🔬 (Sinlge-text version)
25
- Visit our project on Github: [chatgpt-comparison-detection project](https://github.com/Hello-SimpleAI/chatgpt-comparison-detection)<br>
26
- 欢迎在 Github 上关注我们的 [ChatGPT 对比与检测项目](https://github.com/Hello-SimpleAI/chatgpt-comparison-detection)
27
-
28
- We provide three kinds of detectors, all in Bilingual / 我们提供了三个版本的检测器,且都支持中英文:
29
- - [**QA version / 问答版**](https://huggingface.co/spaces/Hello-SimpleAI/chatgpt-detector-qa)<br>
30
- detect whether an **answer** is generated by ChatGPT for certain **question**, using PLM-based classifiers / 判断某个**问题的回答**是否由ChatGPT生成,使用基于PTM的分类器来开发;
31
- - [Sinlge-text version / 独立文本版 (👈 Current / 当前使用)](https://huggingface.co/spaces/Hello-SimpleAI/chatgpt-detector-single)<br>
32
- detect whether a piece of text is ChatGPT generated, using PLM-based classifiers / 判断**单条文本**是否由ChatGPT生成,使用基于PTM的分类器来开发;
33
- - [Linguistic version / 语言学版](https://huggingface.co/spaces/Hello-SimpleAI/chatgpt-detector-ling)<br>
34
- detect whether a piece of text is ChatGPT generated, using linguistic features / 判断**单条文本**是否由ChatGPT生成,使用基于语言学特征的模型来开发;
35
-
36
-
37
  """)
38
- with gr.Tab("English"):
39
- gr.Markdown("""
40
- Note: Providing more text to the `Text` box can make the prediction more accurate!
41
- """)
42
- t1 = gr.Textbox(lines=5, label='Text',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.")
43
- button1 = gr.Button("🤖 Predict!")
44
- label1 = gr.Textbox(lines=1, label='Predicted Label 🎃')
45
- score1 = gr.Textbox(lines=1, label='Prob')
46
  with gr.Tab("中文版"):
47
  gr.Markdown("""
48
  注意: 在`文本`栏中输入更多的文本,可以让预测更准确哦!
@@ -51,13 +32,16 @@ with gr.Blocks() as demo:
51
  button2 = gr.Button("🤖 预测!")
52
  label2 = gr.Textbox(lines=1, label='预测结果 🎃')
53
  score2 = gr.Textbox(lines=1, label='模型概率')
 
 
 
 
 
 
 
 
54
 
55
  button1.click(predict_en, inputs=[t1], outputs=[label1,score1], api_name='predict_en')
56
  button2.click(predict_zh, inputs=[t2], outputs=[label2,score2], api_name='predict_zh')
57
 
58
- # Page Count
59
- gr.Markdown("""
60
- <center><a href='https://clustrmaps.com/site/1bsdc' title='Visit tracker'><img src='//clustrmaps.com/map_v2.png?cl=080808&w=a&t=tt&d=NXQdnwxvIm27veMbB5F7oHNID09nhSvkBRZ_Aji9eIA&co=ffffff&ct=808080'/></a></center>
61
- """)
62
-
63
  demo.launch()
 
2
  import gradio as gr
3
  from transformers import pipeline
4
 
5
+
6
+ pipeline_en = pipeline(task="text-classification", model="Hello-SimpleAI/chatgpt-detector-roberta")
7
+ pipeline_zh = pipeline(task="text-classification", model="Hello-SimpleAI/chatgpt-detector-roberta-chinese")
8
 
9
 
10
 
 
21
 
22
  with gr.Blocks() as demo:
23
  gr.Markdown("""
24
+ ## 🔬🔬🔬
 
 
 
 
 
 
 
 
 
 
 
 
25
  """)
26
+
 
 
 
 
 
 
 
27
  with gr.Tab("中文版"):
28
  gr.Markdown("""
29
  注意: 在`文本`栏中输入更多的文本,可以让预测更准确哦!
 
32
  button2 = gr.Button("🤖 预测!")
33
  label2 = gr.Textbox(lines=1, label='预测结果 🎃')
34
  score2 = gr.Textbox(lines=1, label='模型概率')
35
+ with gr.Tab("English"):
36
+ gr.Markdown("""
37
+ Note: Providing more text to the `Text` box can make the prediction more accurate!
38
+ """)
39
+ t1 = gr.Textbox(lines=5, label='Text',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.")
40
+ button1 = gr.Button("🤖 Predict!")
41
+ label1 = gr.Textbox(lines=1, label='Predicted Label 🎃')
42
+ score1 = gr.Textbox(lines=1, label='Prob')
43
 
44
  button1.click(predict_en, inputs=[t1], outputs=[label1,score1], api_name='predict_en')
45
  button2.click(predict_zh, inputs=[t2], outputs=[label2,score2], api_name='predict_zh')
46
 
 
 
 
 
 
47
  demo.launch()