not-lain commited on
Commit
ace499a
·
verified ·
1 Parent(s): b52c99a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -40,7 +40,7 @@ def get_mask(model, input_img, use_amp=True, s=640):
40
  def process(img):
41
  path = load_img(img,output_type="str")
42
  img = cv2.cvtColor(cv2.imread(path, cv2.IMREAD_COLOR), cv2.COLOR_BGR2RGB)
43
- mask = get_mask(HF_model, img, use_amp= False, s=img_size)
44
  img = mask * img
45
  out = load_img(img)
46
  return out
 
40
  def process(img):
41
  path = load_img(img,output_type="str")
42
  img = cv2.cvtColor(cv2.imread(path, cv2.IMREAD_COLOR), cv2.COLOR_BGR2RGB)
43
+ mask = get_mask(model, img, use_amp= False, s=img_size)
44
  img = mask * img
45
  out = load_img(img)
46
  return out