sergeipetrov commited on
Commit
0af2224
1 Parent(s): 952ebd6

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +1 -1
handler.py CHANGED
@@ -15,7 +15,7 @@ logger = logging.getLogger(__name__)
15
 
16
 
17
  class EndpointHandler():
18
- def __init__(self, model_settings):
19
 
20
  device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
21
  logger.info(f"Using device: {device.type}")
 
15
 
16
 
17
  class EndpointHandler():
18
+ def __init__(self, path=""):
19
 
20
  device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
21
  logger.info(f"Using device: {device.type}")