Upload model.py

model.py

@@ -3,10 +3,10 @@ from transformers import Wav2Vec2BertModel
 
 
 class SpoofVerificationModel(nn.Module):
-    def __init__(self, num_types=59):
+    def __init__(self, w2v_path='facebook/w2v-bert-2.0', num_types=59):
         super(SpoofVerificationModel, self).__init__()
 
-        self.wav2vec2 = None
+        self.wav2vec2 = Wav2Vec2BertModel.from_pretrained(w2v_path, output_hidden_states=True)
         self.wav2vec_config = self.wav2vec2.config
 
         self.deepfake_embed = nn.Linear(self.wav2vec2.config.hidden_size, 1024)