File size: 2,293 Bytes
5f9f1bb
58ab0a6
 
 
 
 
 
 
 
 
 
 
 
 
 
e337991
 
 
 
 
 
 
 
 
 
58ab0a6
 
 
 
 
 
 
fbe14df
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
import torch
from transformers import AutoTokenizer, AutoModelForSequenceClassification
import gradio as gr

tokenizer = AutoTokenizer.from_pretrained("hugfaceguy0001/AITextDetector")
model = AutoModelForSequenceClassification.from_pretrained("hugfaceguy0001/AITextDetector")
def detect_text(text):
    inputs = tokenizer(text, return_tensors="pt")
    logits = model(**inputs).logits
    probs = torch.nn.functional.softmax(logits)[0,:]
    labels = {model.config.id2label[i]:float(probs[i]) for i in range(3)}
    labels = dict(sorted(labels.items(),key=lambda x:x[1],reverse=True))
    return labels

myexamples = ["黑神话这个游戏,你喷它优化、喷它地图、喷它空气墙、喷它战斗、喷它剧情;这些东西不管我自己觉得有没有喷一下的必要,可也都或多或少的能理解它为什么被喷的原因。", 
        "加密通信助长犯罪的说法并不成立,理由如下:隐私权的基本保障:加密技术保障了个人和组织的隐私权,这是基本人权的一部分。无论是个人的日常通信、商业机密,还是新闻记者的来源保护,加密通信都在确保合法和正当的隐私需求不被侵犯。",
             """“我喜欢你,让我产生了风雨的感觉” 这句话表达了一种复杂而强烈的情感。可以从以下几个角度来理解:

1. **感情的起伏**:风雨通常象征着不稳定和变化,表达出喜欢一个人可能带来的情感起伏。喜欢一个人可能会带来喜悦、兴奋,也可能带来不安、焦虑等情感的波动。

2. **情感的冲击**:风雨也可以象征着某种强烈的冲击,喜欢一个人可能会让人内心产生强烈的震撼或心动,就像暴风雨般来得猛烈而不可控制。

3. **不确定性和挑战**:风雨往往意味着不确定和挑战。喜欢一个人可能意味着进入一段未知的感情旅程,充满了挑战和不可预测的因素。

总的来说,这句话形象地表达了喜欢一个人所带来的那种复杂、强烈且充满变化的内心感受。"""]

demo = gr.Interface(fn=detect_text,
             inputs=gr.Text(label="请输入待检测的文本:"),
             outputs=gr.Label(num_top_classes=3),
             examples=myexamples
            )

demo.launch()