Tao Wu commited on
Commit
7a43d21
·
1 Parent(s): c1fc72c

change retriever

Browse files
Files changed (1) hide show
  1. app/config.py +2 -1
app/config.py CHANGED
@@ -6,7 +6,8 @@ REDIS_PORT = int(os.getenv('REDIS_PORT', 6379))
6
  REDIS_DB = int(os.getenv('REDIS_DB', 0))
7
 
8
  # Model and embedding configuration
9
- MODEL_NAME = os.getenv('MODEL_NAME', "intfloat/multilingual-e5-large-instruct")
 
10
  ENCODE_KWARGS = {
11
  'normalize_embeddings': os.getenv('NORMALIZE_EMBEDDINGS', 'True') == 'True',
12
  'convert_to_tensor': os.getenv('CONVERT_TO_TENSOR', 'True') == 'True'
 
6
  REDIS_DB = int(os.getenv('REDIS_DB', 0))
7
 
8
  # Model and embedding configuration
9
+ #MODEL_NAME = os.getenv('MODEL_NAME', "intfloat/multilingual-e5-large-instruct")
10
+ MODEL_NAME = os.getenv('MODEL_NAME', "wt3639/EduGBERT_CourseRec")
11
  ENCODE_KWARGS = {
12
  'normalize_embeddings': os.getenv('NORMALIZE_EMBEDDINGS', 'True') == 'True',
13
  'convert_to_tensor': os.getenv('CONVERT_TO_TENSOR', 'True') == 'True'