Hugging Face's logo

Spaces:
cksleigen
/
ScriptFromChat
Sleeping

App Files Community
ScriptFromChat
File size: 9,677 Bytes 
8a10ea6
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
 
# config.py
from transformers import AutoTokenizer, AutoModelForSequenceClassification

# λͺ¨λΈ 및 ν† ν¬λ‚˜μ΄μ € λ‘œλ“œ
model_name = "hun3359/klue-bert-base-sentiment"
tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModelForSequenceClassification.from_pretrained(model_name)

# 라벨 맀핑
label_mapping = {
    0: 'λΆ„λ…Έ', 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: 'μ΄ˆμ‘°ν•œ',
    30: 'μƒμ²˜', 31: 'μ§ˆνˆ¬ν•˜λŠ”', 32: 'λ°°μ‹ λ‹Ήν•œ', 33: '고립된', 34: '좩격 받은', 35: 'κ°€λ‚œν•œ λΆˆμš°ν•œ', 36: 'ν¬μƒλœ', 37: 'μ–΅μšΈν•œ', 38: 'κ΄΄λ‘œμ›Œν•˜λŠ”', 39: '버렀진',
    40: 'λ‹Ήν™©', 41: '고립된(λ‹Ήν™©ν•œ)', 42: 'λ‚¨μ˜ μ‹œμ„ μ„ μ˜μ‹ν•˜λŠ”', 43: 'μ™Έλ‘œμš΄', 44: '열등감', 45: 'μ£„μ±…κ°μ˜', 46: 'λΆ€λ„λŸ¬μš΄', 47: '혐였슀러운', 48: 'ν•œμ‹¬ν•œ', 49: 'ν˜Όλž€μŠ€λŸ¬μš΄(λ‹Ήν™©ν•œ)',
    50: '기쁨', 51: 'κ°μ‚¬ν•˜λŠ”', 52: 'μ‹ λ’°ν•˜λŠ”', 53: 'νŽΈμ•ˆν•œ', 54: '만쑱슀러운', 55: 'ν₯λΆ„', 56: 'λŠκΈ‹', 57: 'μ•ˆλ„', 58: '신이 λ‚œ', 59: 'μžμ‹ ν•˜λŠ”'
}

# Big5 차원 μ •μ˜
big5_dimensions = ['Openness', 'Conscientiousness', 'Extraversion', 'Agreeableness', 'Neuroticism']

# 감정-Big5 μ—°κ΄€μ„± 사전 μ •μ˜
emotion_big5_priors = {
    # λΆ„λ…Έ κ΄€λ ¨ (0-9)
    'λΆ„λ…Έ': {'Openness': -0.1, 'Conscientiousness': -0.3, 'Extraversion': 0.2, 'Agreeableness': -0.7, 'Neuroticism': 0.8},
    'νˆ΄νˆ΄λŒ€λŠ”': {'Openness': -0.2, 'Conscientiousness': -0.4, 'Extraversion': -0.1, 'Agreeableness': -0.6, 'Neuroticism': 0.7},
    'μ’Œμ ˆν•œ': {'Openness': -0.1, 'Conscientiousness': -0.2, 'Extraversion': -0.3, 'Agreeableness': -0.2, 'Neuroticism': 0.8},
    'μ§œμ¦λ‚΄λŠ”': {'Openness': -0.1, 'Conscientiousness': -0.3, 'Extraversion': 0.1, 'Agreeableness': -0.6, 'Neuroticism': 0.7},
    '방어적인': {'Openness': -0.3, 'Conscientiousness': 0.1, 'Extraversion': -0.2, 'Agreeableness': -0.5, 'Neuroticism': 0.6},
    'μ•…μ˜μ μΈ': {'Openness': -0.2, 'Conscientiousness': -0.4, 'Extraversion': 0.0, 'Agreeableness': -0.8, 'Neuroticism': 0.4},
    'μ•ˆλ‹¬ν•˜λŠ”': {'Openness': 0.0, 'Conscientiousness': -0.2, 'Extraversion': 0.3, 'Agreeableness': -0.3, 'Neuroticism': 0.8},
    'κ΅¬μ—­μ§ˆ λ‚˜λŠ”': {'Openness': -0.1, 'Conscientiousness': 0.0, 'Extraversion': -0.4, 'Agreeableness': -0.3, 'Neuroticism': 0.7},
    'λ…Έμ—¬μ›Œν•˜λŠ”': {'Openness': -0.1, 'Conscientiousness': -0.2, 'Extraversion': 0.2, 'Agreeableness': -0.7, 'Neuroticism': 0.8},
    'μ„±κ°€μ‹ ': {'Openness': -0.2, 'Conscientiousness': -0.3, 'Extraversion': 0.0, 'Agreeableness': -0.5, 'Neuroticism': 0.6},

    # μŠ¬ν”” κ΄€λ ¨ (10-19)
    'μŠ¬ν””': {'Openness': 0.1, 'Conscientiousness': -0.1, 'Extraversion': -0.6, 'Agreeableness': 0.2, 'Neuroticism': 0.8},
    'μ‹€λ§ν•œ': {'Openness': 0.0, 'Conscientiousness': -0.2, 'Extraversion': -0.4, 'Agreeableness': 0.1, 'Neuroticism': 0.7},
    'λΉ„ν†΅ν•œ': {'Openness': 0.2, 'Conscientiousness': -0.1, 'Extraversion': -0.7, 'Agreeableness': 0.3, 'Neuroticism': 0.9},
    'ν›„νšŒλ˜λŠ”': {'Openness': 0.1, 'Conscientiousness': -0.3, 'Extraversion': -0.3, 'Agreeableness': 0.2, 'Neuroticism': 0.7},
    'μš°μšΈν•œ': {'Openness': 0.0, 'Conscientiousness': -0.4, 'Extraversion': -0.8, 'Agreeableness': 0.0, 'Neuroticism': 0.9},
    'λ§ˆλΉ„λœ': {'Openness': -0.2, 'Conscientiousness': -0.5, 'Extraversion': -0.7, 'Agreeableness': 0.0, 'Neuroticism': 0.8},
    '염세적인': {'Openness': 0.0, 'Conscientiousness': -0.3, 'Extraversion': -0.5, 'Agreeableness': -0.2, 'Neuroticism': 0.7},
    '눈물이 λ‚˜λŠ”': {'Openness': 0.2, 'Conscientiousness': 0.0, 'Extraversion': -0.4, 'Agreeableness': 0.3, 'Neuroticism': 0.8},
    'λ‚™λ‹΄ν•œ': {'Openness': 0.0, 'Conscientiousness': -0.3, 'Extraversion': -0.6, 'Agreeableness': 0.1, 'Neuroticism': 0.8},
    'ν™˜λ©Έμ„ λŠλΌλŠ”': {'Openness': 0.1, 'Conscientiousness': -0.2, 'Extraversion': -0.4, 'Agreeableness': -0.3, 'Neuroticism': 0.6},

    # λΆˆμ•ˆ κ΄€λ ¨ (20-29)
    'λΆˆμ•ˆ': {'Openness': 0.0, 'Conscientiousness': -0.1, 'Extraversion': -0.4, 'Agreeableness': 0.1, 'Neuroticism': 0.9},
    'λ‘λ €μš΄': {'Openness': -0.1, 'Conscientiousness': 0.0, 'Extraversion': -0.5, 'Agreeableness': 0.2, 'Neuroticism': 0.8},
    '슀트레슀 λ°›λŠ”': {'Openness': 0.0, 'Conscientiousness': -0.2, 'Extraversion': -0.3, 'Agreeableness': -0.1, 'Neuroticism': 0.9},
    'μ·¨μ•½ν•œ': {'Openness': 0.1, 'Conscientiousness': -0.3, 'Extraversion': -0.6, 'Agreeableness': 0.3, 'Neuroticism': 0.8},
    'ν˜Όλž€μŠ€λŸ¬μš΄': {'Openness': -0.1, 'Conscientiousness': -0.4, 'Extraversion': -0.2, 'Agreeableness': 0.0, 'Neuroticism': 0.7},
    'λ‹Ήν˜ΉμŠ€λŸ¬μš΄': {'Openness': -0.2, 'Conscientiousness': -0.2, 'Extraversion': -0.3, 'Agreeableness': 0.1, 'Neuroticism': 0.6},
    '회의적인': {'Openness': 0.2, 'Conscientiousness': 0.1, 'Extraversion': -0.2, 'Agreeableness': -0.3, 'Neuroticism': 0.4},
    'κ±±μ •μŠ€λŸ¬μš΄': {'Openness': 0.0, 'Conscientiousness': 0.2, 'Extraversion': -0.3, 'Agreeableness': 0.3, 'Neuroticism': 0.8},
    'μ‘°μ‹¬μŠ€λŸ¬μš΄': {'Openness': -0.1, 'Conscientiousness': 0.4, 'Extraversion': -0.4, 'Agreeableness': 0.2, 'Neuroticism': 0.5},
    'μ΄ˆμ‘°ν•œ': {'Openness': 0.0, 'Conscientiousness': -0.1, 'Extraversion': 0.1, 'Agreeableness': -0.1, 'Neuroticism': 0.8},

    # μƒμ²˜ κ΄€λ ¨ (30-39)
    'μƒμ²˜': {'Openness': 0.1, 'Conscientiousness': 0.0, 'Extraversion': -0.4, 'Agreeableness': 0.2, 'Neuroticism': 0.8},
    'μ§ˆνˆ¬ν•˜λŠ”': {'Openness': 0.0, 'Conscientiousness': -0.2, 'Extraversion': 0.1, 'Agreeableness': -0.6, 'Neuroticism': 0.7},
    'λ°°μ‹ λ‹Ήν•œ': {'Openness': 0.0, 'Conscientiousness': 0.1, 'Extraversion': -0.3, 'Agreeableness': -0.2, 'Neuroticism': 0.8},
    '고립된': {'Openness': 0.0, 'Conscientiousness': -0.1, 'Extraversion': -0.8, 'Agreeableness': 0.1, 'Neuroticism': 0.7},
    '좩격 받은': {'Openness': 0.1, 'Conscientiousness': -0.1, 'Extraversion': -0.3, 'Agreeableness': 0.0, 'Neuroticism': 0.8},
    'κ°€λ‚œν•œ λΆˆμš°ν•œ': {'Openness': 0.0, 'Conscientiousness': -0.2, 'Extraversion': -0.4, 'Agreeableness': 0.2, 'Neuroticism': 0.6},
    'ν¬μƒλœ': {'Openness': 0.1, 'Conscientiousness': 0.2, 'Extraversion': -0.2, 'Agreeableness': 0.4, 'Neuroticism': 0.6},
    'μ–΅μšΈν•œ': {'Openness': 0.0, 'Conscientiousness': 0.1, 'Extraversion': -0.1, 'Agreeableness': -0.2, 'Neuroticism': 0.7},
    'κ΄΄λ‘œμ›Œν•˜λŠ”': {'Openness': 0.1, 'Conscientiousness': -0.1, 'Extraversion': -0.5, 'Agreeableness': 0.2, 'Neuroticism': 0.9},
    '버렀진': {'Openness': 0.0, 'Conscientiousness': -0.2, 'Extraversion': -0.7, 'Agreeableness': 0.1, 'Neuroticism': 0.8},

    # λ‹Ήν™© κ΄€λ ¨ (40-49)
    'λ‹Ήν™©': {'Openness': -0.1, 'Conscientiousness': -0.2, 'Extraversion': -0.4, 'Agreeableness': 0.1, 'Neuroticism': 0.7},
    '고립된(λ‹Ήν™©ν•œ)': {'Openness': -0.1, 'Conscientiousness': -0.1, 'Extraversion': -0.8, 'Agreeableness': 0.0, 'Neuroticism': 0.7},
    'λ‚¨μ˜ μ‹œμ„ μ„ μ˜μ‹ν•˜λŠ”': {'Openness': -0.2, 'Conscientiousness': 0.1, 'Extraversion': -0.3, 'Agreeableness': 0.2, 'Neuroticism': 0.6},
    'μ™Έλ‘œμš΄': {'Openness': 0.0, 'Conscientiousness': 0.0, 'Extraversion': -0.8, 'Agreeableness': 0.1, 'Neuroticism': 0.7},
    '열등감': {'Openness': -0.1, 'Conscientiousness': -0.3, 'Extraversion': -0.5, 'Agreeableness': 0.0, 'Neuroticism': 0.8},
    'μ£„μ±…κ°μ˜': {'Openness': 0.1, 'Conscientiousness': 0.3, 'Extraversion': -0.2, 'Agreeableness': 0.4, 'Neuroticism': 0.7},
    'λΆ€λ„λŸ¬μš΄': {'Openness': -0.1, 'Conscientiousness': 0.0, 'Extraversion': -0.5, 'Agreeableness': 0.2, 'Neuroticism': 0.6},
    '혐였슀러운': {'Openness': -0.2, 'Conscientiousness': 0.0, 'Extraversion': -0.2, 'Agreeableness': -0.5, 'Neuroticism': 0.5},
    'ν•œμ‹¬ν•œ': {'Openness': -0.2, 'Conscientiousness': -0.3, 'Extraversion': -0.4, 'Agreeableness': -0.2, 'Neuroticism': 0.6},
    'ν˜Όλž€μŠ€λŸ¬μš΄(λ‹Ήν™©ν•œ)': {'Openness': -0.2, 'Conscientiousness': -0.3, 'Extraversion': -0.3, 'Agreeableness': 0.0, 'Neuroticism': 0.7},

    # 기쁨 κ΄€λ ¨ (50-59)
    '기쁨': {'Openness': 0.3, 'Conscientiousness': 0.2, 'Extraversion': 0.7, 'Agreeableness': 0.4, 'Neuroticism': -0.6},
    'κ°μ‚¬ν•˜λŠ”': {'Openness': 0.2, 'Conscientiousness': 0.3, 'Extraversion': 0.3, 'Agreeableness': 0.7, 'Neuroticism': -0.4},
    'μ‹ λ’°ν•˜λŠ”': {'Openness': 0.1, 'Conscientiousness': 0.4, 'Extraversion': 0.2, 'Agreeableness': 0.8, 'Neuroticism': -0.5},
    'νŽΈμ•ˆν•œ': {'Openness': 0.1, 'Conscientiousness': 0.3, 'Extraversion': 0.1, 'Agreeableness': 0.3, 'Neuroticism': -0.7},
    '만쑱슀러운': {'Openness': 0.2, 'Conscientiousness': 0.4, 'Extraversion': 0.3, 'Agreeableness': 0.2, 'Neuroticism': -0.6},
    'ν₯λΆ„': {'Openness': 0.4, 'Conscientiousness': 0.0, 'Extraversion': 0.8, 'Agreeableness': 0.1, 'Neuroticism': 0.2},
    'λŠκΈ‹': {'Openness': 0.2, 'Conscientiousness': -0.2, 'Extraversion': 0.1, 'Agreeableness': 0.3, 'Neuroticism': -0.8},
    'μ•ˆλ„': {'Openness': 0.0, 'Conscientiousness': 0.1, 'Extraversion': 0.0, 'Agreeableness': 0.2, 'Neuroticism': -0.7},
    '신이 λ‚œ': {'Openness': 0.3, 'Conscientiousness': 0.0, 'Extraversion': 0.8, 'Agreeableness': 0.3, 'Neuroticism': -0.3},
    'μžμ‹ ν•˜λŠ”': {'Openness': 0.2, 'Conscientiousness': 0.5, 'Extraversion': 0.6, 'Agreeableness': 0.1, 'Neuroticism': -0.6}
}