meisaicheck-api / config.py
vumichien's picture
Update model configuration to use OpenVINO, modify cached embeddings with new sizes, and enhance .gitignore to exclude additional data files for improved project management.
58b07a2
raw
history blame
1.5 kB
import os
# Security Config
SECRET_KEY = "09d25e094faa6ca2556c818166b7a9563b93f7099f6f0f4caa6cf63b88e8d3e7"
ALGORITHM = "HS256"
ACCESS_TOKEN_EXPIRE_HOURS = 24
# Simple API Key (for HF Spaces)
API_KEY = "meisai-api-key-2025"
API_KEY_NAME = "X-API-KEY"
# Paths
BASE_DIR = os.path.dirname(os.path.abspath(__file__))
DATA_DIR = os.path.join(BASE_DIR, "data")
UPLOAD_DIR = os.path.join(BASE_DIR, "uploads")
OUTPUT_DIR = os.path.join(BASE_DIR, "outputs")
SUBJECT_DATA_FILE = os.path.join(DATA_DIR, "subjectData.csv")
SAMPLE_DATA_FILE = os.path.join(DATA_DIR, "sampleData.csv")
STANDARD_NAME_MAP_DATA_FILE = os.path.join(DATA_DIR, "standardData.csv")
ABSTRACT_MAP_DATA_FILE = os.path.join(DATA_DIR, "abstractMapData.csv")
NAME_ABSTRACT_MAP_DATA_FILE = os.path.join(DATA_DIR, "nameAbstractData.csv")
# Model Config
HALF = "float16"
MODEL_TYPE = "openvino"
DEVICE_TYPE = "cpu"
MODEL_NAME = "Detomo/cl-nagoya-sup-simcse-ja-nss-v1_0_8_5"
if HALF == "float16":
SENTENCE_EMBEDDING_FILE = os.path.join(
DATA_DIR,
"anchor_name_sentence_embeddings_cl-nagoya-sup-simcse-ja-nss-v_1_0_7_13_float16.pkl",
)
elif HALF == "bfloat16":
SENTENCE_EMBEDDING_FILE = os.path.join(
DATA_DIR,
"anchor_name_sentence_embeddings_cl-nagoya-sup-simcse-ja-nss-v_1_0_7_13_bfloat16.pkl",
)
else:
SENTENCE_EMBEDDING_FILE = os.path.join(
DATA_DIR,
"anchor_name_sentence_embeddings_cl-nagoya-sup-simcse-ja-nss-v_1_0_7_13.pkl",
)
# SENTENCE_EMBEDDING_FILE = None