Spaces:
Running
Running
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()
|