File size: 1,239 Bytes
b77c0a2
 
 
 
 
 
 
 
 
 
 
 
 
 
c164ba9
c41415e
 
 
c1cf1c2
892f887
c41415e
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
import os

# Security Config
SECRET_KEY = "09d25e094faa6ca2556c818166b7a9563b93f7099f6f0f4caa6cf63b88e8d3e7"
ALGORITHM = "HS256"
ACCESS_TOKEN_EXPIRE_HOURS = 24

# 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")

# Model Config
HALF = "float16"
MODEL_TYPE = None
DEVICE_TYPE = "cpu"
MODEL_NAME = "Detomo/cl-nagoya-sup-simcse-ja-nss-v_1_0_6"
if HALF == "float16":
    SENTENCE_EMBEDDING_FILE = os.path.join(
        DATA_DIR,
        "anchor_name_sentence_embeddings_cl-nagoya-sup-simcse-ja-nss-v_1_0_6_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_6_bfloat16.pkl",
    )
else:
    SENTENCE_EMBEDDING_FILE = os.path.join(
        DATA_DIR,
        "anchor_name_sentence_embeddings_cl-nagoya-sup-simcse-ja-nss-v_1_0_6.pkl",
    )
# SENTENCE_EMBEDDING_FILE = None