Spaces:
Running
on
Zero
Running
on
Zero
Upload model.py
Browse files
model.py
CHANGED
@@ -3,10 +3,10 @@ from transformers import Wav2Vec2BertModel
|
|
3 |
|
4 |
|
5 |
class SpoofVerificationModel(nn.Module):
|
6 |
-
def __init__(self,
|
7 |
super(SpoofVerificationModel, self).__init__()
|
8 |
|
9 |
-
self.wav2vec2 =
|
10 |
self.wav2vec_config = self.wav2vec2.config
|
11 |
|
12 |
self.deepfake_embed = nn.Linear(self.wav2vec2.config.hidden_size, 1024)
|
|
|
3 |
|
4 |
|
5 |
class SpoofVerificationModel(nn.Module):
|
6 |
+
def __init__(self, num_types=59):
|
7 |
super(SpoofVerificationModel, self).__init__()
|
8 |
|
9 |
+
self.wav2vec2 = None
|
10 |
self.wav2vec_config = self.wav2vec2.config
|
11 |
|
12 |
self.deepfake_embed = nn.Linear(self.wav2vec2.config.hidden_size, 1024)
|