cdcvd commited on
Commit
f99f042
·
verified ·
1 Parent(s): 170b0d7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -67,6 +67,8 @@ def process_file(input_file):
67
 
68
  input_folder = 'output_images'
69
  for i, img in enumerate(images):
 
 
70
  img.save(os.path.join(input_folder, f'image_{i}.jpg'))
71
 
72
  return images
 
67
 
68
  input_folder = 'output_images'
69
  for i, img in enumerate(images):
70
+ if img.mode == 'RGBA':
71
+ img = img.convert('RGB')
72
  img.save(os.path.join(input_folder, f'image_{i}.jpg'))
73
 
74
  return images