KhaldiAbderrhmane
/

emotion-classifier-hubert

Text Classification

Model card Files Files and versions Community

KhaldiAbderrhmane commited on Jun 24

Commit

528a5f2

•

1 Parent(s): d181956

Update modeling_emotion_classifier.py

Files changed (1) hide show

modeling_emotion_classifier.py +1 -1

modeling_emotion_classifier.py CHANGED Viewed

@@ -1,7 +1,7 @@
 import os
 import torch
 import torch.nn as nn
-from transformers import PreTrainedModel, HubertConfig, HubertModel
 from transformers.file_utils import (
     WEIGHTS_NAME,
     TF2_WEIGHTS_NAME,

 import os
 import torch
 import torch.nn as nn
+from transformers import PreTrainedModel, HubertConfig, HubertModel,PretrainedConfig
 from transformers.file_utils import (
     WEIGHTS_NAME,
     TF2_WEIGHTS_NAME,