frodos
/

dis-background-removal

Inference Endpoints

Model card Files Files and versions Community

frodos commited on Apr 7

Commit

99e5048

•

1 Parent(s): e9311f3

Fix handler

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -26,7 +26,7 @@ def get_model(device="cpu"):
 class EndpointHandler():
-    def __init__(self):
         self._model = get_model()
     def __call__(self, data: Dict[str, Any]) -> list[Dict[str, Any]]:

 class EndpointHandler():
+    def __init__(self, path=""):
         self._model = get_model()
     def __call__(self, data: Dict[str, Any]) -> list[Dict[str, Any]]: