shir0ha commited on
Commit
e8d559d
1 Parent(s): 2dcd1be

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ model.eval()
14
  def predict(inp):
15
  print("***********Inference****************")
16
  res = inference_img(model, inp)
17
- print(res.shape, res.type)
18
  # main_subject = inp.copy()
19
  # main_subject = main_subject * res[res > 0.1]
20
  return main_subject
 
14
  def predict(inp):
15
  print("***********Inference****************")
16
  res = inference_img(model, inp)
17
+ print(res.shape, res.dtype)
18
  # main_subject = inp.copy()
19
  # main_subject = main_subject * res[res > 0.1]
20
  return main_subject