shivansh-ka's picture
testing
f467969
raw
history blame contribute delete
307 Bytes
import os
ROOT_DIR = os.getcwd()
MODEL_DIR_NAME = "serving_model"
MODEL_NAME = "roberta-fine-tuned-2"
MODEL_PATH = os.path.join(ROOT_DIR, MODEL_DIR_NAME, MODEL_NAME)
TOKENIZER_FILE_NAME = "tokenizer"
TOKENIZER_PATH = os.path.join(ROOT_DIR, MODEL_DIR_NAME, TOKENIZER_FILE_NAME)
MAX_LEN =192
BUFFER_SIZE=2048