Wataru commited on
Commit
6209e88
·
1 Parent(s): c9914a1
Files changed (1) hide show
  1. app.py +0 -2
app.py CHANGED
@@ -23,9 +23,7 @@ def load_model(checkpoint_path: str):
23
  global model
24
  model = SSLMOSLightningModule.load_from_checkpoint(
25
  checkpoint_path, map_location=device,
26
- ssl_model_path='',
27
  pretrained_model_path=None,
28
- ssl_model_type='w2v2',
29
  )
30
  model.eval()
31
  print(f"Model loaded from {checkpoint_path}")
 
23
  global model
24
  model = SSLMOSLightningModule.load_from_checkpoint(
25
  checkpoint_path, map_location=device,
 
26
  pretrained_model_path=None,
 
27
  )
28
  model.eval()
29
  print(f"Model loaded from {checkpoint_path}")