prasadnu commited on
Commit
3cdaefa
·
1 Parent(s): dc0e654

updated ML models

Browse files
pages/Semantic_Search.py CHANGED
@@ -265,10 +265,10 @@ if "SAGEMAKER_CrossEncoder_MODEL_ID" not in st.session_state:
265
 
266
 
267
  if "SAGEMAKER_SPARSE_MODEL_ID" not in st.session_state:
268
- st.session_state.SAGEMAKER_SPARSE_MODEL_ID = "fkol-ZMBTp0efWqBcO2P"
269
 
270
  if "BEDROCK_TEXT_MODEL_ID" not in st.session_state:
271
- st.session_state.BEDROCK_TEXT_MODEL_ID = "usQk-ZMBkiQuoz1QFmXN"
272
  #bytes_for_rekog = ""
273
  bedrock_ = boto3.client('bedrock-runtime',
274
  aws_access_key_id=st.secrets['user_access_key'],
 
265
 
266
 
267
  if "SAGEMAKER_SPARSE_MODEL_ID" not in st.session_state:
268
+ st.session_state.SAGEMAKER_SPARSE_MODEL_ID = "rWKMoZcBXWKpawCjlOPr"
269
 
270
  if "BEDROCK_TEXT_MODEL_ID" not in st.session_state:
271
+ st.session_state.BEDROCK_TEXT_MODEL_ID = "p2KFoZcBXWKpawCjzON4"
272
  #bytes_for_rekog = ""
273
  bedrock_ = boto3.client('bedrock-runtime',
274
  aws_access_key_id=st.secrets['user_access_key'],
semantic_search/all_search_execute.py CHANGED
@@ -152,7 +152,7 @@ def handler(input_,session_id):
152
  ######### Create the queries for hybrid search #########
153
 
154
 
155
- path = "demostore-search-index-reindex/_search"
156
 
157
  url = host + path
158
 
@@ -413,7 +413,7 @@ def handler(input_,session_id):
413
  del hybrid_payload["query"]["hybrid"]
414
  hybrid_payload["query"] = single_query
415
  if(st.session_state.re_ranker == 'true' and st.session_state.input_reranker == 'Cohere Rerank'):
416
- path = "demostore-search-index-reindex/_search?search_pipeline=rerank_pipeline"
417
  url = host + path
418
  hybrid_payload["ext"] = {"rerank": {
419
  "query_context": {
 
152
  ######### Create the queries for hybrid search #########
153
 
154
 
155
+ path = "demostore-search-index-reindex-new/_search"
156
 
157
  url = host + path
158
 
 
413
  del hybrid_payload["query"]["hybrid"]
414
  hybrid_payload["query"] = single_query
415
  if(st.session_state.re_ranker == 'true' and st.session_state.input_reranker == 'Cohere Rerank'):
416
+ path = "demostore-search-index-reindex-new/_search?search_pipeline=rerank_pipeline"
417
  url = host + path
418
  hybrid_payload["ext"] = {"rerank": {
419
  "query_context": {