prasadnu commited on
Commit
10f24a5
·
1 Parent(s): 1f43c77

rerank model

Browse files
Files changed (1) hide show
  1. utilities/invoke_models.py +6 -2
utilities/invoke_models.py CHANGED
@@ -12,11 +12,15 @@ import streamlit as st
12
 
13
  region = 'us-east-1'
14
  @st.cache_resource
15
- bedrock_runtime_client = boto3.client(
 
16
  'bedrock-runtime',
17
  aws_access_key_id=st.secrets['user_access_key'],
18
  aws_secret_access_key=st.secrets['user_secret_key'], region_name = 'us-east-1'
19
- )
 
 
 
20
 
21
 
22
 
 
12
 
13
  region = 'us-east-1'
14
  @st.cache_resource
15
+ def get_bedrock_client():
16
+ return boto3.client(
17
  'bedrock-runtime',
18
  aws_access_key_id=st.secrets['user_access_key'],
19
  aws_secret_access_key=st.secrets['user_secret_key'], region_name = 'us-east-1'
20
+ )
21
+
22
+
23
+ bedrock_runtime_client = get_bedrock_client():
24
 
25
 
26