File size: 3,649 Bytes
8cb40c4
 
 
 
 
 
 
 
 
 
 
 
 
387f42d
 
 
8cb40c4
 
 
 
 
 
 
 
 
 
387f42d
8cb40c4
387f42d
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
8cb40c4
387f42d
 
 
 
8cb40c4
387f42d
 
 
 
 
 
8cb40c4
387f42d
8cb40c4
 
387f42d
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
8cb40c4
 
387f42d
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
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
import torch
import gradio as gr
from transformers import (
    AutoTokenizer,
    AutoModelForSequenceClassification
)

MODEL_NAME = "duclo90/results"
tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
model = AutoModelForSequenceClassification.from_pretrained(MODEL_NAME)
model.eval()

def classify_text(text):
    if not text.strip():
        return "โš ๏ธ Please enter some text to analyze."
    
    inputs = tokenizer(
        text,
        return_tensors="pt",
        truncation=True,
        max_length=512
    )
    with torch.no_grad():
        outputs = model(**inputs)
        probs = torch.softmax(outputs.logits, dim=-1)
        score, pred = torch.max(probs, dim=-1)
    
    label = model.config.id2label[pred.item()]
    confidence = round(score.item() * 100, 1)
    
    # Format output with icons and styling
    icon = "๐Ÿ‘ค" if label.lower() == "human" else "๐Ÿค–"
    
    result = f"""
    ### {icon} Classification Result
    
    **Prediction:** {label}-Written Text
    
    **Confidence:** {confidence}%
    
    {'๐ŸŸฆ' * int(confidence // 10)} {confidence}%
    """
    
    return result

# Custom CSS for modern look
custom_css = """
#component-0 {
    max-width: 900px;
    margin: auto;
    padding: 20px;
}

.gradio-container {
    background: linear-gradient(135deg, #667eea 0%, #764ba2 100%) !important;
    font-family: 'Inter', sans-serif;
}

#title {
    text-align: center;
    color: white;
    font-size: 2.5em;
    font-weight: 700;
    margin-bottom: 10px;
    text-shadow: 2px 2px 4px rgba(0,0,0,0.3);
}

#description {
    text-align: center;
    color: rgba(255,255,255,0.9);
    font-size: 1.1em;
    margin-bottom: 30px;
}

.input-textarea, .output-markdown {
    border-radius: 15px !important;
    border: 2px solid rgba(255,255,255,0.3) !important;
    background: rgba(255,255,255,0.95) !important;
    box-shadow: 0 8px 32px rgba(0,0,0,0.1) !important;
}

button {
    background: linear-gradient(135deg, #f093fb 0%, #f5576c 100%) !important;
    border: none !important;
    border-radius: 10px !important;
    padding: 12px 30px !important;
    font-weight: 600 !important;
    font-size: 16px !important;
    box-shadow: 0 4px 15px rgba(0,0,0,0.2) !important;
    transition: transform 0.2s !important;
}

button:hover {
    transform: translateY(-2px) !important;
    box-shadow: 0 6px 20px rgba(0,0,0,0.3) !important;
}

.footer {
    display: none !important;
}
"""

# Example texts
examples = [
    ["The sun set over the horizon, painting the sky in brilliant shades of orange and pink. I sat there, mesmerized by nature's beauty."],
    ["According to recent data analysis, the implementation of artificial intelligence systems has increased operational productivity by approximately 40 percent across various industries."],
    ["Can't believe how amazing that concert was last night! The energy in the crowd was absolutely electric ๐ŸŽธโœจ"],
]

# Create interface with modern theme
iface = gr.Interface(
    fn=classify_text,
    inputs=gr.Textbox(
        lines=8, 
        placeholder="โœ๏ธ Paste or type text here to analyze...",
        label="Enter Text",
        elem_id="input-text"
    ),
    outputs=gr.Markdown(label="Analysis Result"),
    title="<div id='title'>๐Ÿ” AI Text Classifier</div>",
    description="<div id='description'>Detect whether text is human-written or AI-generated using advanced NLP</div>",
    examples=examples,
    theme=gr.themes.Soft(
        primary_hue="purple",
        secondary_hue="pink",
        neutral_hue="slate",
        font=gr.themes.GoogleFont("Inter")
    ),
    css=custom_css,
    flagging_mode="never"
)

iface.launch()