young476 commited on
Commit
51a6dc6
·
verified ·
1 Parent(s): 983e5c2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -8,8 +8,8 @@ from transformers import AutoTokenizer, BertForSequenceClassification
8
  # --- 1. 설정 및 전처리 함수 ---
9
  device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
10
 
11
- BASE_TOKENIZER_DIR = 'base_tokenizer'
12
- EMOTION_MODEL_DIR = 'kobert_emotion_classifier_archive'
13
  GENRE_MODEL_DIR = 'kobert_genre_classifier_archive'
14
 
15
  okt = Okt()
 
8
  # --- 1. 설정 및 전처리 함수 ---
9
  device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
10
 
11
+ BASE_TOKENIZER_DIR = 'base'
12
+ EMOTION_MODEL_DIR = 'kobert_emotion_classifier'
13
  GENRE_MODEL_DIR = 'kobert_genre_classifier_archive'
14
 
15
  okt = Okt()