Commit
·
6dd9966
1
Parent(s):
a2238f4
Add inference endpoint feature.
Browse files- 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 |
)
|