Spaces:
Sleeping
Sleeping
Update src/streamlit_app.py
Browse files- src/streamlit_app.py +6 -2
src/streamlit_app.py
CHANGED
|
@@ -34,11 +34,15 @@ st.set_page_config(
|
|
| 34 |
layout="wide",
|
| 35 |
)
|
| 36 |
|
| 37 |
-
|
| 38 |
-
|
|
|
|
|
|
|
|
|
|
| 39 |
VECTOR_PATH = DATA_DIR / "vector_store.index"
|
| 40 |
PKL_PATH = DATA_DIR / "data.pkl"
|
| 41 |
|
|
|
|
| 42 |
EMBED_MODEL_NAME = os.getenv("EMBED_MODEL_NAME", "BAAI/bge-large-en-v1.5")
|
| 43 |
TOP_K = int(os.getenv("TOP_K", "5"))
|
| 44 |
MAX_CONTEXT_CHARS = int(os.getenv("MAX_CONTEXT_CHARS", "12000"))
|
|
|
|
| 34 |
layout="wide",
|
| 35 |
)
|
| 36 |
|
| 37 |
+
from pathlib import Path
|
| 38 |
+
|
| 39 |
+
BASE_DIR = Path(__file__).resolve().parent # points to src/
|
| 40 |
+
DATA_DIR = BASE_DIR / "data"
|
| 41 |
+
|
| 42 |
VECTOR_PATH = DATA_DIR / "vector_store.index"
|
| 43 |
PKL_PATH = DATA_DIR / "data.pkl"
|
| 44 |
|
| 45 |
+
|
| 46 |
EMBED_MODEL_NAME = os.getenv("EMBED_MODEL_NAME", "BAAI/bge-large-en-v1.5")
|
| 47 |
TOP_K = int(os.getenv("TOP_K", "5"))
|
| 48 |
MAX_CONTEXT_CHARS = int(os.getenv("MAX_CONTEXT_CHARS", "12000"))
|