sergeipetrov
/

asrdiarization-handler

Inference Endpoints

Model card Files Files and versions Community

sergeipetrov HF staff commited on Mar 28

Commit

0af2224

•

1 Parent(s): 952ebd6

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -15,7 +15,7 @@ logger = logging.getLogger(__name__)
 class EndpointHandler():
-    def __init__(self, model_settings):
         device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
         logger.info(f"Using device: {device.type}")

 class EndpointHandler():
+    def __init__(self, path=""):
         device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
         logger.info(f"Using device: {device.type}")