nielsr HF staff commited on
Commit
9f112a2
1 Parent(s): 4e9eaed

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -24,7 +24,7 @@ def process_image(image):
24
  size=image.size[::-1],
25
  mode="bicubic",
26
  align_corners=False,
27
- )
28
  output = prediction.cpu().numpy()
29
  formatted = (output * 255 / np.max(output)).astype('uint8')
30
  img = Image.fromarray(formatted)
 
24
  size=image.size[::-1],
25
  mode="bicubic",
26
  align_corners=False,
27
+ ).squeeze()
28
  output = prediction.cpu().numpy()
29
  formatted = (output * 255 / np.max(output)).astype('uint8')
30
  img = Image.fromarray(formatted)