Added back path parameter in case it has to be in the signature
Browse files- handler.py +1 -1
handler.py
CHANGED
@@ -14,7 +14,7 @@ def _fake_generate(n: int = 3):
|
|
14 |
|
15 |
|
16 |
class EndpointHandler():
|
17 |
-
def __init__(self, use_cuda: bool =
|
18 |
# Preload all the elements you are going to need at inference.
|
19 |
# pseudo:
|
20 |
# self.model= load_model(path)
|
|
|
14 |
|
15 |
|
16 |
class EndpointHandler():
|
17 |
+
def __init__(self, path="", use_cuda: bool = True, test_mode: bool= False):
|
18 |
# Preload all the elements you are going to need at inference.
|
19 |
# pseudo:
|
20 |
# self.model= load_model(path)
|