Luigi commited on
Commit
22699db
·
1 Parent(s): 641e003

fix device type error

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -102,7 +102,7 @@ def get_diarization_pipe():
102
  use_auth_token=token or True
103
  )
104
 
105
- dar_pipe.to("cpu")
106
  return dar_pipe
107
 
108
 
 
102
  use_auth_token=token or True
103
  )
104
 
105
+ dar_pipe.to(torch.device("cpu"))
106
  return dar_pipe
107
 
108