m97j commited on
Commit
7b7bcd6
·
1 Parent(s): 216aedb

Initial commit

Browse files
Files changed (1) hide show
  1. config.py +7 -0
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")