befozg commited on
Commit
14cca43
·
1 Parent(s): fa90067

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -16,7 +16,7 @@ def predict(inp):
16
  print("***********Inference****************")
17
  mask = inference_img(model, inp)
18
  inp_np = np.array(inp)
19
- fg = np.uint8((mask[:,:,None]*inp_np).numpy())
20
  print("***********Inference finish****************")
21
  print("***********MASK****************", inp_np.max(), mask.max())
22
 
 
16
  print("***********Inference****************")
17
  mask = inference_img(model, inp)
18
  inp_np = np.array(inp)
19
+ fg = np.uint8((mask[:,:,None]*inp_np))
20
  print("***********Inference finish****************")
21
  print("***********MASK****************", inp_np.max(), mask.max())
22