shikibu9419 commited on
Commit
5513895
·
1 Parent(s): b1ef674

Fix full color bug

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -131,6 +131,7 @@ async def optimize(
131
  (opt_img, preview_img) = optimize_img(process_name, img_array)
132
 
133
  h, w = preview_img.shape[:2]
 
134
  if process_name.endswith('full'):
135
  opt_img = np.reshape(opt_img, (h, w, 3))
136
  else:
 
131
  (opt_img, preview_img) = optimize_img(process_name, img_array)
132
 
133
  h, w = preview_img.shape[:2]
134
+ preview_img = np.reshape(preview_img, (h, w, 3))
135
  if process_name.endswith('full'):
136
  opt_img = np.reshape(opt_img, (h, w, 3))
137
  else: