specify ssl model type

app.py

@@ -21,7 +21,8 @@ def load_model(checkpoint_path: str):
     """Loads the model from the given checkpoint path."""
     global model
     model = SSLMOSLightningModule.load_from_checkpoint(
-        checkpoint_path, map_location=device
+        checkpoint_path, map_location=device,
+        ssl_model_type='w2v2',
     )
     model.eval()
     print(f"Model loaded from {checkpoint_path}")