File size: 1,094 Bytes
75db241
 
 
9e66cf0
75db241
 
d4c8507
 
 
 
 
 
 
75db241
 
d4c8507
 
 
 
75db241
 
 
 
 
 
 
 
 
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
import gradio as gr
from transformers import pipeline

model_name = "ZombitX64/Thai-sentiment-e5"  # เปลี่ยนเป็นชื่อโมเดลของคุณ
nlp = pipeline("sentiment-analysis", model=model_name)

# สร้าง dict แม็ป label เป็นเลข
label_map = {
    "negative": 0,
    "neutral": 1,
    "positive": 2
}

def analyze_text(text):
    result = nlp(text)[0]
    label = result['label'].lower()  # แปลงเป็น lowercase เผื่อโมเดลให้แบบนี้
    score = result['score']
    code = label_map.get(label, -1)  # ถ้า label ไม่อยู่ใน dict ให้เป็น -1
    return f"ผลวิเคราะห์: {label} (รหัส: {code}) ความมั่นใจ {score:.2f}"

demo = gr.Interface(
    fn=analyze_text,
    inputs=gr.Textbox(lines=3, placeholder="พิมพ์ข้อความที่นี่..."),
    outputs="text",
    title="แอปวิเคราะห์ข้อความภาษาไทย"
)

demo.launch()