s3 bucket

src/predict.py

@@ -83,6 +83,8 @@ class Predictor:
         self.s3_client = boto3.client('s3',aws_access_key_id=os.getenv('AWS_ACCESS_KEY'), aws_secret_access_key=os.getenv('AWS_SECRET_KEY'))
         print(os.getenv("AWS_ACCESS_KEY"))
         print(os.getenv("AWS_SECRET_KEY"))
+        self.bucket_name = os.getenv('S3_BUCKET_NAME')
+        print(self.bucket_name)
 
 
     def setup(self):
@@ -150,7 +152,7 @@ class Predictor:
         header = file_buffer.read(12)
         print(header)
         file_format = None
-        bucket_name = 'demovidelyuseruploads'
+        bucket_name = self.bucket_name
         if b'WAVE' in header:
             file_format = 'wav'
         elif b'OggS' in header:
@@ -220,7 +222,7 @@ class Predictor:
         
         
         s3_key = s3_url.split('/')[-1]
-        bucket_name = 'demovidelyuseruploads'
+        bucket_name = self.bucket_name
         local_file_path = os.path.join(raw_dir,s3_key)
         self.download_file_from_s3(self.s3_client,bucket_name,s3_key,local_file_path)
         #voice_clone with styletts2
@@ -275,7 +277,7 @@ class Predictor:
         
         
         s3_key = s3_url.split('/')[-1]
-        bucket_name = 'demovidelyuseruploads'
+        bucket_name = self.bucket_name
         local_file_path = os.path.join(raw_dir,s3_key)
         self.download_file_from_s3(self.s3_client,bucket_name,s3_key,local_file_path)
         #voice_clone with styletts2
@@ -317,7 +319,7 @@ class Predictor:
         
         
         s3_key = s3_url.split('/')[-1]
-        bucket_name = 'demovidelyuseruploads'
+        bucket_name = self.bucket_name
         local_file_path = os.path.join(raw_dir,s3_key)
         self.download_file_from_s3(self.s3_client,bucket_name,s3_key,local_file_path)
         model,sampler = self.model,self.sampler