cszhzleo commited on
Commit
50e8e05
1 Parent(s): 55f2708

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -0
app.py CHANGED
@@ -9,6 +9,13 @@ sm_endpoint_name = os.getenv("sm_endpoint_name")
9
  access_key = os.getenv("access_key")
10
  secret_key = os.getenv("secret_key")
11
 
 
 
 
 
 
 
 
12
  # hyperparameters for llm
13
  parameters = {
14
  "do_sample": True,
 
9
  access_key = os.getenv("access_key")
10
  secret_key = os.getenv("secret_key")
11
 
12
+ session = boto3.Session(
13
+ aws_access_key_id=access_key,
14
+ aws_secret_access_key=secret_key,
15
+ region_name=region
16
+ )
17
+ sess = sagemaker.Session(boto_session=session)
18
+
19
  # hyperparameters for llm
20
  parameters = {
21
  "do_sample": True,