Nattapong Tapachoom commited on
Commit
20959f5
·
1 Parent(s): a2c4342

Add new sentiment model to MODEL_LIST and remove unused pipeline initialization

Browse files
Files changed (1) hide show
  1. app.py +2 -6
app.py CHANGED
@@ -1,7 +1,6 @@
1
  import gradio as gr
2
  from transformers import pipeline
3
 
4
- <<<<<<< HEAD
5
 
6
  # รายชื่อโมเดลที่ให้เลือก
7
  MODEL_LIST = [
@@ -16,7 +15,8 @@ MODEL_LIST = [
16
  "ZombitX64/MultiSent-E5-Pro",
17
  "ZombitX64/MultiSent-E5",
18
  "ZombitX64/Thai-sentiment-e5",
19
- "ZombitX64/sentiment-103"
 
20
  ]
21
 
22
  from functools import lru_cache
@@ -25,10 +25,6 @@ from functools import lru_cache
25
  @lru_cache(maxsize=2)
26
  def get_nlp(model_name):
27
  return pipeline("sentiment-analysis", model=model_name)
28
- =======
29
- model_name = "ZombitX64/Sentiment-03"
30
- nlp = pipeline("sentiment-analysis", model=model_name)
31
- >>>>>>> 9fff20c2f551dc64f34e70b76c0606f173cd3464
32
 
33
  label_map = {
34
  "LABEL_0": 0,
 
1
  import gradio as gr
2
  from transformers import pipeline
3
 
 
4
 
5
  # รายชื่อโมเดลที่ให้เลือก
6
  MODEL_LIST = [
 
15
  "ZombitX64/MultiSent-E5-Pro",
16
  "ZombitX64/MultiSent-E5",
17
  "ZombitX64/Thai-sentiment-e5",
18
+ "ZombitX64/sentiment-103",
19
+ "nlptown/bert-base-multilingual-uncased-sentiment"
20
  ]
21
 
22
  from functools import lru_cache
 
25
  @lru_cache(maxsize=2)
26
  def get_nlp(model_name):
27
  return pipeline("sentiment-analysis", model=model_name)
 
 
 
 
28
 
29
  label_map = {
30
  "LABEL_0": 0,