Spaces:

sarulab-speech
/

MSR-UTMOS

Running on Zero

Wataru commited on 26 days ago

Commit

c9914a1

1 Parent(s): 70722dd

fix

Files changed (1) hide show

app.py CHANGED Viewed

@@ -23,7 +23,8 @@ def load_model(checkpoint_path: str):
     global model
     model = SSLMOSLightningModule.load_from_checkpoint(
         checkpoint_path, map_location=device,
-        pretrained_model_path='',
         ssl_model_type='w2v2',
     )
     model.eval()

     global model
     model = SSLMOSLightningModule.load_from_checkpoint(
         checkpoint_path, map_location=device,
+        ssl_model_path='',
+        pretrained_model_path=None,
         ssl_model_type='w2v2',
     )
     model.eval()