Update app.py
Browse files
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")
|