sergeipetrov
/

asrdiarization-handler

Inference Endpoints

Model card Files Files and versions Community

sergeipetrov commited on Mar 28

Commit

6b7e6e6

•

1 Parent(s): 36cfeae

Update handler.py

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -26,7 +26,7 @@ class InferenceConfig(BaseModel):
 class EndpointHandler():
-        def __init__(self):
         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):
         device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
         logger.info(f"Using device: {device.type}")