fthor commited on
Commit
5909e50
1 Parent(s): 4b86165

Added back path parameter in case it has to be in the signature

Browse files
Files changed (1) hide show
  1. 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 = False, test_mode: bool= False):
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)