hyeongnym commited on
Commit
833313d
1 Parent(s): 4d3af82

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -0
app.py CHANGED
@@ -419,3 +419,12 @@ class TestChatBot(unittest.TestCase):
419
  test_query = "테스트 질문"
420
  result = find_relevant_context(test_query)
421
  self.assertIsInstance(result, list)
 
 
 
 
 
 
 
 
 
 
419
  test_query = "테스트 질문"
420
  result = find_relevant_context(test_query)
421
  self.assertIsInstance(result, list)
422
+ class Config:
423
+ def __init__(self):
424
+ # 변경: EleutherAI/polyglot-ko-12.8b 모델 사용
425
+ self.MODEL_ID = "EleutherAI/polyglot-ko-12.8b" # 한국어 특화 모델
426
+ self.MAX_HISTORY = 10
427
+ self.MAX_TOKENS = 4096
428
+ self.DEFAULT_TEMPERATURE = 0.8
429
+ self.HF_TOKEN = os.environ.get("HF_TOKEN", None)
430
+ self.MODELS = os.environ.get("MODELS")