Spaces:
Runtime error
Runtime error
Update models.py
Browse files
models.py
CHANGED
@@ -11,7 +11,7 @@ from transformers.models.hubert.modeling_hubert import (
|
|
11 |
HubertModel
|
12 |
)
|
13 |
|
14 |
-
from
|
15 |
|
16 |
|
17 |
class Wav2Vec2ClassificationHead(nn.Module):
|
|
|
11 |
HubertModel
|
12 |
)
|
13 |
|
14 |
+
from modeling_outputs import SpeechClassifierOutput
|
15 |
|
16 |
|
17 |
class Wav2Vec2ClassificationHead(nn.Module):
|