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")
|