kootaeng2 commited on
Commit
c1b9543
ยท
1 Parent(s): 96b3478

Chore: Improve and update code comments

Browse files
scripts/save_complete_model.py CHANGED
@@ -1,4 +1,6 @@
1
  # save_complete_model.py
 
 
2
  from transformers import AutoModelForSequenceClassification, AutoTokenizer
3
 
4
  # ๊ธฐ์กด ํ›ˆ๋ จ ๊ฒฐ๊ณผ๋ฌผ์ด ์ €์žฅ๋œ ๊ฒฝ๋กœ
 
1
  # save_complete_model.py
2
+ # ํ›ˆ๋ จ๋œ ๋ชจ๋ธ์„ ๋ฐฐํฌ์šฉ์œผ๋กœ ๋ณ€ํ™˜
3
+
4
  from transformers import AutoModelForSequenceClassification, AutoTokenizer
5
 
6
  # ๊ธฐ์กด ํ›ˆ๋ จ ๊ฒฐ๊ณผ๋ฌผ์ด ์ €์žฅ๋œ ๊ฒฝ๋กœ
scripts/upload_model.py CHANGED
@@ -1,4 +1,5 @@
1
  # upload_model.py
 
2
 
3
  from transformers import AutoTokenizer, AutoModelForSequenceClassification
4
 
 
1
  # upload_model.py
2
+ # ๋ฐฐํฌ์šฉ ๋ชจ๋ธ์„ Hugging Face Hub์— ์—…๋กœ๋“œ
3
 
4
  from transformers import AutoTokenizer, AutoModelForSequenceClassification
5
 
src/emotion_engine.py CHANGED
@@ -1,5 +1,4 @@
1
- # emotion_engine.py (์ˆ˜์ • ํ›„ ์ตœ์ข… ๋ฒ„์ „)
2
-
3
  import torch
4
  from transformers import AutoTokenizer, AutoModelForSequenceClassification, pipeline
5
  import os
@@ -9,7 +8,6 @@ from transformers import AutoTokenizer, AutoModelForSequenceClassification, pipe
9
  import os
10
 
11
  def load_emotion_classifier():
12
- # --- ์ด ๋ถ€๋ถ„์„ ์ˆ˜์ •ํ•ฉ๋‹ˆ๋‹ค ---
13
  # ๋กœ์ปฌ ๊ฒฝ๋กœ ๋Œ€์‹ , Hugging Face Hub์˜ ๋ชจ๋ธ ID๋ฅผ ์‚ฌ์šฉํ•ฉ๋‹ˆ๋‹ค.
14
  MODEL_PATH = "koons/korean-emotion-classifier-final" # "์‚ฌ์šฉ์ž์ด๋ฆ„/๋ชจ๋ธ์ด๋ฆ„" ํ˜•์‹
15
 
@@ -30,7 +28,6 @@ def load_emotion_classifier():
30
 
31
  return emotion_classifier
32
 
33
- # predict_emotion ํ•จ์ˆ˜๋Š” ๊ทธ๋Œ€๋กœ ๋‘ก๋‹ˆ๋‹ค.
34
  def predict_emotion(classifier, text):
35
  if not text or not text.strip(): return "๋‚ด์šฉ ์—†์Œ"
36
  if classifier is None: return "์˜ค๋ฅ˜: ๊ฐ์ • ๋ถ„์„ ์—”์ง„ ์ค€๋น„ ์•ˆ๋จ."
 
1
+ # emotion_engine.py
 
2
  import torch
3
  from transformers import AutoTokenizer, AutoModelForSequenceClassification, pipeline
4
  import os
 
8
  import os
9
 
10
  def load_emotion_classifier():
 
11
  # ๋กœ์ปฌ ๊ฒฝ๋กœ ๋Œ€์‹ , Hugging Face Hub์˜ ๋ชจ๋ธ ID๋ฅผ ์‚ฌ์šฉํ•ฉ๋‹ˆ๋‹ค.
12
  MODEL_PATH = "koons/korean-emotion-classifier-final" # "์‚ฌ์šฉ์ž์ด๋ฆ„/๋ชจ๋ธ์ด๋ฆ„" ํ˜•์‹
13
 
 
28
 
29
  return emotion_classifier
30
 
 
31
  def predict_emotion(classifier, text):
32
  if not text or not text.strip(): return "๋‚ด์šฉ ์—†์Œ"
33
  if classifier is None: return "์˜ค๋ฅ˜: ๊ฐ์ • ๋ถ„์„ ์—”์ง„ ์ค€๋น„ ์•ˆ๋จ."