chace_dir eklendi
Browse files- src/app.py +2 -1
src/app.py
CHANGED
|
@@ -15,6 +15,7 @@ TEXT_COLS = ['STRA', 'STRB']
|
|
| 15 |
LABEL_COL = 'DISTANCE'
|
| 16 |
EXCLUDE_COLS = TEXT_COLS + [LABEL_COL, 'FILLER']
|
| 17 |
NUMERICAL_FEATURE_DIM = 5132
|
|
|
|
| 18 |
|
| 19 |
@st.cache_data
|
| 20 |
def load_data():
|
|
@@ -27,7 +28,7 @@ def load_data():
|
|
| 27 |
|
| 28 |
@st.cache_resource
|
| 29 |
def load_models_and_tokenizer():
|
| 30 |
-
tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
|
| 31 |
|
| 32 |
cross_encoder_model = tf.keras.models.load_model(
|
| 33 |
SAVED_CROSS_ENCODER_MODEL_PATH,
|
|
|
|
| 15 |
LABEL_COL = 'DISTANCE'
|
| 16 |
EXCLUDE_COLS = TEXT_COLS + [LABEL_COL, 'FILLER']
|
| 17 |
NUMERICAL_FEATURE_DIM = 5132
|
| 18 |
+
CACHE_DIR = "./.cache"
|
| 19 |
|
| 20 |
@st.cache_data
|
| 21 |
def load_data():
|
|
|
|
| 28 |
|
| 29 |
@st.cache_resource
|
| 30 |
def load_models_and_tokenizer():
|
| 31 |
+
tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME, cache_dir=CACHE_DIR)
|
| 32 |
|
| 33 |
cross_encoder_model = tf.keras.models.load_model(
|
| 34 |
SAVED_CROSS_ENCODER_MODEL_PATH,
|