sky4432 commited on
Commit
6dc20d1
Β·
verified Β·
1 Parent(s): 7aff10e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -5,14 +5,15 @@ from transformers import pipeline
5
  # 1) 감성뢄석 λͺ¨λΈ 뢈러였기
6
  sentiment_analyzer = pipeline(
7
  "sentiment-analysis",
8
- model="monologg/koelectra-base-finetuned-nsmc"
 
9
  )
10
 
11
  st.title("읡λͺ…κ²Œμ‹œνŒ 감성뢄석")
12
 
13
  st.write("""
14
  - CSV λ˜λŠ” Excel νŒŒμΌμ„ μ—…λ‘œλ“œν•΄μ£Όμ„Έμš”.
15
- - 'content' μ—΄(λ˜λŠ” μ›ν•˜λŠ” μ—΄)에 κ²Œμ‹œκΈ€ ν…μŠ€νŠΈκ°€ μžˆλ‹€κ³  κ°€μ •ν•©λ‹ˆλ‹€.
16
  - μ—…λ‘œλ“œ ν›„ '감성뢄석 μ‹€ν–‰' λ²„νŠΌμ„ λˆ„λ₯΄λ©΄, 각 글에 λŒ€ν•œ 긍정/λΆ€μ • λ ˆμ΄λΈ”κ³Ό μ μˆ˜κ°€ ν‘œμ‹œλ©λ‹ˆλ‹€.
17
  """)
18
 
@@ -36,7 +37,7 @@ if uploaded_file is not None:
36
 
37
  for text in df["λ‚΄μš©"]:
38
  # 감성뢄석
39
- result = sentiment_analyzer(text)
40
  label = result[0]['label'] # positive/negative
41
  score = result[0]['score'] # 신뒰도(0~1)
42
  results_label.append(label)
 
5
  # 1) 감성뢄석 λͺ¨λΈ 뢈러였기
6
  sentiment_analyzer = pipeline(
7
  "sentiment-analysis",
8
+ model="monologg/koelectra-base-finetuned-nsmc",
9
+ truncation=True
10
  )
11
 
12
  st.title("읡λͺ…κ²Œμ‹œνŒ 감성뢄석")
13
 
14
  st.write("""
15
  - CSV λ˜λŠ” Excel νŒŒμΌμ„ μ—…λ‘œλ“œν•΄μ£Όμ„Έμš”.
16
+ - 'λ‚΄μš©' μ—΄(λ˜λŠ” μ›ν•˜λŠ” μ—΄)에 κ²Œμ‹œκΈ€ ν…μŠ€νŠΈκ°€ μžˆλ‹€κ³  κ°€μ •ν•©λ‹ˆλ‹€.
17
  - μ—…λ‘œλ“œ ν›„ '감성뢄석 μ‹€ν–‰' λ²„νŠΌμ„ λˆ„λ₯΄λ©΄, 각 글에 λŒ€ν•œ 긍정/λΆ€μ • λ ˆμ΄λΈ”κ³Ό μ μˆ˜κ°€ ν‘œμ‹œλ©λ‹ˆλ‹€.
18
  """)
19
 
 
37
 
38
  for text in df["λ‚΄μš©"]:
39
  # 감성뢄석
40
+ result = sentiment_analyzer(text, truncation=True)
41
  label = result[0]['label'] # positive/negative
42
  score = result[0]['score'] # 신뒰도(0~1)
43
  results_label.append(label)