code-slicer commited on
Commit
f4851ab
Β·
verified Β·
1 Parent(s): 30eb937

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +13 -2
app.py CHANGED
@@ -152,9 +152,20 @@ from chat_a import (
152
  )
153
 
154
  # ──────────────────────────────── LLM ────────────────────────────────
155
- OLLAMA_HOST = os.getenv("OLLAMA_HOST", "http://localhost:11434")
 
 
 
 
 
 
 
 
 
 
 
156
  OLLAMA_MODEL = os.getenv("OLLAMA_MODEL", "gemma2:9b")
157
- OLLAMA_TIMEOUT = int(os.getenv("OLLAMA_TIMEOUT", "60"))
158
 
159
  KOREAN_SYSTEM_PROMPT = """당신은 ν•œκ΅­μ–΄ μ–΄μ‹œμŠ€ν„΄νŠΈμž…λ‹ˆλ‹€. 항상 ν•œκ΅­μ–΄λ‘œ λ‹΅ν•˜μ„Έμš”."""
160
 
 
152
  )
153
 
154
  # ──────────────────────────────── LLM ────────────────────────────────
155
+ OLLAMA_HOST = (
156
+ os.getenv("OLLAMA_HOST")
157
+ or st.secrets.get("OLLAMA_HOST", None)
158
+ )
159
+
160
+ # λ§ˆμ§€λ§‰ μ•ˆμ „μž₯치: 미섀정이면 였λ₯˜ μ•ˆλ‚΄
161
+ if not OLLAMA_HOST:
162
+ st.error("OLLAMA_HOSTκ°€ μ„€μ •λ˜μ§€ μ•Šμ•˜μŠ΅λ‹ˆλ‹€. (예: https://<username>-<ollama-space>.hf.space)")
163
+ st.stop()
164
+
165
+ OLLAMA_HOST = OLLAMA_HOST.rstrip("/")
166
+ OLLAMA_TIMEOUT = int(os.getenv("OLLAMA_TIMEOUT", "120"))
167
  OLLAMA_MODEL = os.getenv("OLLAMA_MODEL", "gemma2:9b")
168
+
169
 
170
  KOREAN_SYSTEM_PROMPT = """당신은 ν•œκ΅­μ–΄ μ–΄μ‹œμŠ€ν„΄νŠΈμž…λ‹ˆλ‹€. 항상 ν•œκ΅­μ–΄λ‘œ λ‹΅ν•˜μ„Έμš”."""
171