alfrds commited on
Commit
c3413fa
·
1 Parent(s): d8b212a
Files changed (1) hide show
  1. app.py +16 -14
app.py CHANGED
@@ -22,19 +22,6 @@ pipe = StableDiffusionPipeline.from_pretrained(
22
 
23
  pipe = pipe.to(device)
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/'
37
-
38
 
39
  def text_to_image(prompt, save_as, key_id):
40
 
@@ -44,7 +31,8 @@ def text_to_image(prompt, save_as, key_id):
44
  # Create an instance of the S3 client
45
  s3 = boto3.client('s3',
46
  aws_access_key_id=AWS_ACCESS_KEY_ID,
47
- aws_secret_access_key=AWS_SECRET_ACCESS_KEY)
 
48
 
49
  image_name = '-'.join(save_as.split()) + ".webp"
50
 
@@ -81,6 +69,12 @@ def text_to_image(prompt, save_as, key_id):
81
 
82
  def check_if_exist(bucket_name, key):
83
 
 
 
 
 
 
 
84
  try:
85
  s3.Object(bucket_name, key).load()
86
  except botocore.exceptions.ClientError as e:
@@ -96,6 +90,14 @@ def check_if_exist(bucket_name, key):
96
 
97
  def list_s3_files(bucket_name, folder):
98
 
 
 
 
 
 
 
 
 
99
  my_bucket = s3.Bucket(bucket_name)
100
 
101
  for objects in my_bucket.objects.filter(Prefix=folder):
 
22
 
23
  pipe = pipe.to(device)
24
 
 
 
 
 
 
 
 
 
 
 
 
 
 
25
 
26
  def text_to_image(prompt, save_as, key_id):
27
 
 
31
  # Create an instance of the S3 client
32
  s3 = boto3.client('s3',
33
  aws_access_key_id=AWS_ACCESS_KEY_ID,
34
+ aws_secret_access_key=AWS_SECRET_ACCESS_KEY,
35
+ aws_session_token=AWS_SESSION_TOKEN)
36
 
37
  image_name = '-'.join(save_as.split()) + ".webp"
38
 
 
69
 
70
  def check_if_exist(bucket_name, key):
71
 
72
+ s3_client = boto3.client('s3',
73
+ aws_access_key_id=AWS_ACCESS_KEY_ID,
74
+ aws_secret_access_key=AWS_SECRET_ACCESS_KEY,
75
+ aws_session_token=AWS_SESSION_TOKEN)
76
+
77
+
78
  try:
79
  s3.Object(bucket_name, key).load()
80
  except botocore.exceptions.ClientError as e:
 
90
 
91
  def list_s3_files(bucket_name, folder):
92
 
93
+ s3 = boto3.resource('s3',
94
+ aws_access_key_id=AWS_ACCESS_KEY_ID,
95
+ aws_secret_access_key=AWS_SECRET_ACCESS_KEY,
96
+ aws_session_token=AWS_SESSION_TOKEN)
97
+
98
+ bucket_name = 'pineblogs101145-dev'
99
+ folder = 'public/mdx/'
100
+
101
  my_bucket = s3.Bucket(bucket_name)
102
 
103
  for objects in my_bucket.objects.filter(Prefix=folder):