Dusit-P commited on
Commit
aa409de
·
verified ·
1 Parent(s): 4096fb9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +13 -6
app.py CHANGED
@@ -14,9 +14,13 @@ REPO_ID = os.getenv("REPO_ID", "Dusit-P/thai-sentiment")
14
  DEFAULT_MODEL = os.getenv("DEFAULT_MODEL", "WCB")
15
  HF_TOKEN = os.getenv("HF_TOKEN", None)
16
 
17
- AVAILABLE_CHOICES = ["WCB", "WCB_BiLSTM", "WCB_CNN_BiLSTM", "WCB_4Layer_BiLSTM"]
 
 
 
 
18
  if DEFAULT_MODEL not in AVAILABLE_CHOICES:
19
- DEFAULT_MODEL = "WCB"
20
 
21
  NEG_COLOR = "#F87171"
22
  POS_COLOR = "#34D399"
@@ -529,7 +533,7 @@ with gr.Blocks(title="Thai Sentiment Analysis", theme=gr.themes.Soft()) as demo:
529
  choices=AVAILABLE_CHOICES,
530
  value=DEFAULT_MODEL,
531
  label="🤖 เลือกโมเดล",
532
- info="แนะนำ: WCB สำหรับความเร็ว, WCB_4Layer_BiLSTM สำหรับความแม่นยำ"
533
  )
534
 
535
  # =================== Tab 1: วิเคราะห์หลายรีวิว ===================
@@ -691,10 +695,13 @@ with gr.Blocks(title="Thai Sentiment Analysis", theme=gr.themes.Soft()) as demo:
691
 
692
  **WangchanBERTa Variants** - โมเดล BERT ภาษาไทยที่ได้รับการฝึกสำหรับงานวิเคราะห์ความรู้สึก
693
 
694
- - **WCB**: เร็ว เหมาะกับงานทั่วไป
695
- - **WCB_BiLSTM**: เพิ่มความแม่นยำด้วย BiLSTM
 
 
696
  - **WCB_CNN_BiLSTM**: ใช้ CNN + BiLSTM เพิ่มประสิทธิภาพ
697
- - **WCB_4Layer_BiLSTM**: แม่นยำสูงสุด แต่ช้ากว่า
 
698
 
699
  📌 **หมายเหตุ:** โมเดลวิเคราะห์เฉพาะ **เชิงบวก/เชิงลบ** เท่านั้น (ไม่มี neutral)
700
  """)
 
14
  DEFAULT_MODEL = os.getenv("DEFAULT_MODEL", "WCB")
15
  HF_TOKEN = os.getenv("HF_TOKEN", None)
16
 
17
+ # เลือกเฉพาะโมเดลที่ให้ผลดีที่สุด
18
+ AVAILABLE_CHOICES = ["WCB", "WCB_BiLSTM"]
19
+ # โมเดลที่ซ่อนไว้ (uncomment เพื่อเปิดใช้):
20
+ # AVAILABLE_CHOICES = ["WCB", "WCB_BiLSTM", "WCB_CNN_BiLSTM", "WCB_4Layer_BiLSTM"]
21
+
22
  if DEFAULT_MODEL not in AVAILABLE_CHOICES:
23
+ DEFAULT_MODEL = "WCB_BiLSTM" # เปลี่ยน default เป็นตัวที่ดีที่สุด
24
 
25
  NEG_COLOR = "#F87171"
26
  POS_COLOR = "#34D399"
 
533
  choices=AVAILABLE_CHOICES,
534
  value=DEFAULT_MODEL,
535
  label="🤖 เลือกโมเดล",
536
+ info="WCB = เร็ว | WCB_BiLSTM = แม่นยำสูงสุด (แนะนำ)"
537
  )
538
 
539
  # =================== Tab 1: วิเคราะห์หลายรีวิว ===================
 
695
 
696
  **WangchanBERTa Variants** - โมเดล BERT ภาษาไทยที่ได้รับการฝึกสำหรับงานวิเคราะห์ความรู้สึก
697
 
698
+ - **WCB**: รุ่นพื้นฐาน - เร็ว เหมาะกับงานทั่วไป
699
+ - **WCB_BiLSTM**: เพิ่ม BiLSTM layer - **แม่นยำสูงสุด (แนะนำ)** ⭐
700
+
701
+ <!-- โมเดลอื่นๆ ที่ซ่อนไว้:
702
  - **WCB_CNN_BiLSTM**: ใช้ CNN + BiLSTM เพิ่มประสิทธิภาพ
703
+ - **WCB_4Layer_BiLSTM**: BiLSTM 4 ชั้น (ช้ากว่า)
704
+ -->
705
 
706
  📌 **หมายเหตุ:** โมเดลวิเคราะห์เฉพาะ **เชิงบวก/เชิงลบ** เท่านั้น (ไม่มี neutral)
707
  """)