vumichien commited on
Commit
3f7ef09
1 Parent(s): c62c7e5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -19,7 +19,7 @@ def infer(original_image):
19
  output_image = output_image.reshape(
20
  (np.shape(output_image)[0], np.shape(output_image)[1], 3)
21
  )
22
- output_image = np.uint8(output_image)
23
  return output_image
24
 
25
  iface = gr.Interface(
 
19
  output_image = output_image.reshape(
20
  (np.shape(output_image)[0], np.shape(output_image)[1], 3)
21
  )
22
+ output_image = np.uint32(output_image)
23
  return output_image
24
 
25
  iface = gr.Interface(