ZhengPeng7 commited on
Commit
6df6515
·
verified ·
1 Parent(s): 44f1c96

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -157,7 +157,7 @@ def predict(images, resolution, weights_file):
157
  save_file_path = os.path.join(save_dir, "{}.png".format(os.path.splitext(os.path.basename(image_src))[0]))
158
  image_masked.save(save_file_path)
159
  save_paths.append(save_file_path)
160
- image_masked = cv2.cvtColor(np.array(image_masked.convert('RGB')), cv2.COLOR_RGB2BGR)
161
 
162
  if tab_is_batch:
163
  zip_file_path = os.path.join(save_dir, "{}.zip".format(save_dir))
 
157
  save_file_path = os.path.join(save_dir, "{}.png".format(os.path.splitext(os.path.basename(image_src))[0]))
158
  image_masked.save(save_file_path)
159
  save_paths.append(save_file_path)
160
+ image_masked = np.array(image_masked.convert('RGB'))
161
 
162
  if tab_is_batch:
163
  zip_file_path = os.path.join(save_dir, "{}.zip".format(save_dir))