alfrds commited on
Commit
d8b212a
·
1 Parent(s): 4603c75
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -11,6 +11,7 @@ from time import sleep
11
 
12
  AWS_ACCESS_KEY_ID = os.getenv("AWS_ACCESS_KEY_ID")
13
  AWS_SECRET_ACCESS_KEY = os.getenv("AWS_SECRET_ACCESS_KEY")
 
14
  S3_BUCKET_NAME = 'pineblogs101145-dev'
15
 
16
  model_id = "CompVis/stable-diffusion-v1-4"
@@ -23,11 +24,13 @@ pipe = pipe.to(device)
23
 
24
  s3 = boto3.resource('s3',
25
  aws_access_key_id=AWS_ACCESS_KEY_ID,
26
- aws_secret_access_key=AWS_SECRET_ACCESS_KEY)
 
27
 
28
  s3_client = boto3.client('s3',
29
  aws_access_key_id=AWS_ACCESS_KEY_ID,
30
- aws_secret_access_key=AWS_SECRET_ACCESS_KEY)
 
31
 
32
  bucket_name = 'pineblogs101145-dev'
33
  folder = 'public/mdx/'
 
11
 
12
  AWS_ACCESS_KEY_ID = os.getenv("AWS_ACCESS_KEY_ID")
13
  AWS_SECRET_ACCESS_KEY = os.getenv("AWS_SECRET_ACCESS_KEY")
14
+ AWS_SESSION_TOKEN = os.getenv("AWS_SESSION_TOKEN")
15
  S3_BUCKET_NAME = 'pineblogs101145-dev'
16
 
17
  model_id = "CompVis/stable-diffusion-v1-4"
 
24
 
25
  s3 = boto3.resource('s3',
26
  aws_access_key_id=AWS_ACCESS_KEY_ID,
27
+ aws_secret_access_key=AWS_SECRET_ACCESS_KEY,
28
+ aws_session_token=AWS_SESSION_TOKEN)
29
 
30
  s3_client = boto3.client('s3',
31
  aws_access_key_id=AWS_ACCESS_KEY_ID,
32
+ aws_secret_access_key=AWS_SECRET_ACCESS_KEY,
33
+ aws_session_token=AWS_SESSION_TOKEN)
34
 
35
  bucket_name = 'pineblogs101145-dev'
36
  folder = 'public/mdx/'