prasadnu commited on
Commit
97e72c0
·
1 Parent(s): 7c3f23a
Files changed (1) hide show
  1. utilities/mvectors.py +2 -1
utilities/mvectors.py CHANGED
@@ -6,7 +6,8 @@ import streamlit as st
6
  import boto3
7
  import json
8
 
9
- runtime = boto3.client('sagemaker-runtime',region_name='us-east-1')
 
10
  # Load model from HuggingFace Hub
11
  tokenizer = AutoTokenizer.from_pretrained('sentence-transformers/all-MiniLM-L6-v2')
12
  endpoint_name = 'huggingface-pytorch-inference-2025-05-21-16-31-07-967'
 
6
  import boto3
7
  import json
8
 
9
+ runtime = boto3.client('sagemaker-runtime',aws_access_key_id=st.secrets['user_access_key'],
10
+ aws_secret_access_key=st.secrets['user_secret_key'],region_name='us-east-1')
11
  # Load model from HuggingFace Hub
12
  tokenizer = AutoTokenizer.from_pretrained('sentence-transformers/all-MiniLM-L6-v2')
13
  endpoint_name = 'huggingface-pytorch-inference-2025-05-21-16-31-07-967'