ZhengPeng7 commited on
Commit
6dd9966
·
1 Parent(s): a2238f4

Add inference endpoint feature.

Browse files
Files changed (1) hide show
  1. handler.py +1 -1
handler.py CHANGED
@@ -87,7 +87,7 @@ else:
87
 
88
 
89
  class EndpointHandler():
90
- def __init__(self):
91
  self.birefnet = AutoModelForImageSegmentation.from_pretrained(
92
  '/'.join(('zhengpeng7', usage_to_weights_file[usage])), trust_remote_code=True
93
  )
 
87
 
88
 
89
  class EndpointHandler():
90
+ def __init__(self, path=''):
91
  self.birefnet = AutoModelForImageSegmentation.from_pretrained(
92
  '/'.join(('zhengpeng7', usage_to_weights_file[usage])), trust_remote_code=True
93
  )