generativeai dannop commited on
Commit
9ebca05
1 Parent(s): a169ad2

Update app.py (#1)

Browse files

- Update app.py (37797fc0e43bb4bd2bfd8c01a84dcccca717c3fc)


Co-authored-by: Daniel <dannop@users.noreply.huggingface.co>

Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -12,7 +12,9 @@ def check_image_similarity(photo_shoot_id):
12
 
13
  images = []
14
  for file in files:
15
- images.append(AwsService.get_image_from_s3(os.environ.get('AWS_S3_BUCKET'), file['Key']))
 
 
16
 
17
  return ImageSimilarity(1).check(images)
18
 
 
12
 
13
  images = []
14
  for file in files:
15
+ params = file['Key'].split('/')
16
+ if params[-1] != '' and params[-1] != ' ':
17
+ images.append(AwsService.get_image_from_s3(os.environ.get('AWS_S3_BUCKET'), file['Key']))
18
 
19
  return ImageSimilarity(1).check(images)
20