Spaces:
Running
on
T4
Running
on
T4
mvectors
Browse files- 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',
|
|
|
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'
|