Spaces:
Running
on
T4
Running
on
T4
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 = "
|
269 |
|
270 |
if "BEDROCK_TEXT_MODEL_ID" not in st.session_state:
|
271 |
-
st.session_state.BEDROCK_TEXT_MODEL_ID = "
|
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": {
|