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):
|