StormblessedKal commited on
Commit
7b7efd0
1 Parent(s): 1887d9b
Files changed (1) hide show
  1. src/predict.py +1 -1
src/predict.py CHANGED
@@ -153,7 +153,7 @@ class Predictor:
153
  parsed_url = urlparse(s3_url)
154
  s3_key = parsed_url.path.lstrip('/')
155
  local_filename = f"{uuid.uuid4()}"
156
- self.download_file_from_s3(bucket_name, s3_key, local_filename)
157
  with open(local_filename, 'rb') as file:
158
  file_bytes = file.read()
159
  os.remove(local_filename)
 
153
  parsed_url = urlparse(s3_url)
154
  s3_key = parsed_url.path.lstrip('/')
155
  local_filename = f"{uuid.uuid4()}"
156
+ self.download_file_from_s3(self.s3_client,bucket_name, s3_key, local_filename)
157
  with open(local_filename, 'rb') as file:
158
  file_bytes = file.read()
159
  os.remove(local_filename)