leonelhs commited on
Commit
eb90f17
·
1 Parent(s): b7b8aab

fix remove background

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -44,7 +44,9 @@ def enhance_light(image):
44
 
45
 
46
  def zero_background(image, new_bgr=None):
47
- return clientZeroBackground.predict(image, new_bgr, api_name="/predict")[0]
 
 
48
 
49
 
50
  def parse_face(image):
 
44
 
45
 
46
  def zero_background(image, new_bgr=None):
47
+ # return clientZeroBackground.predict(image, new_bgr, api_name="/predict")[0]
48
+ # Fixme: cant find predict funciton by name
49
+ return clientZeroBackground.predict(image, new_bgr, fn_index=0)[1]
50
 
51
 
52
  def parse_face(image):