Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
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": {
|