Spaces:
Sleeping
Sleeping
Initial commit
Browse files
config.py
CHANGED
|
@@ -6,6 +6,13 @@ BASE_DIR = Path(__file__).resolve().parent # ai_server/
|
|
| 6 |
# Hugging Face Token
|
| 7 |
HF_TOKEN = os.getenv("HF_TOKEN")
|
| 8 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 9 |
# 모델 이름
|
| 10 |
EMOTION_MODEL_NAME = os.getenv("EMOTION_MODEL_NAME", "tae898/emoberta-base-ko")
|
| 11 |
FALLBACK_MODEL_NAME = os.getenv("FALLBACK_MODEL_NAME", "skt/ko-gpt-trinity-1.2B-v0.5")
|
|
|
|
| 6 |
# Hugging Face Token
|
| 7 |
HF_TOKEN = os.getenv("HF_TOKEN")
|
| 8 |
|
| 9 |
+
# Hugging Face Serve URL
|
| 10 |
+
HF_SERVE_URL = os.getenv("HF_SERVE_URL", "https://m97j-personachatengine-hf-serve.hf.space/api")
|
| 11 |
+
|
| 12 |
+
# Hugging Face Serve Timeout (초)
|
| 13 |
+
HF_TIMEOUT = float(os.getenv("HF_TIMEOUT", "25"))
|
| 14 |
+
|
| 15 |
+
|
| 16 |
# 모델 이름
|
| 17 |
EMOTION_MODEL_NAME = os.getenv("EMOTION_MODEL_NAME", "tae898/emoberta-base-ko")
|
| 18 |
FALLBACK_MODEL_NAME = os.getenv("FALLBACK_MODEL_NAME", "skt/ko-gpt-trinity-1.2B-v0.5")
|