shir0ha commited on
Commit
81c7e37
1 Parent(s): 9ee0378

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -15,9 +15,9 @@ 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 res
21
 
22
  print("MODEL LOADED")
23
  print("************************************")
 
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 > 0.1)
20
+ return main_subject
21
 
22
  print("MODEL LOADED")
23
  print("************************************")