Spaces:

Dddrl
/

emotion_detection_test

Sleeping

Dddrl commited on Mar 26

Commit

f280e49

verified ·

1 Parent(s): 38de4bc

Upload 2 files

Files changed (1) hide show

app.py CHANGED Viewed

@@ -64,7 +64,7 @@ class CNN(nn.Module):
         return x
 model = CNN(5)
-model.load_state_dict(torch.load("best_model_CNN_bs32_lr0.0005_epoch12_acc0.9248.pth", map_location=torch.device("cpu")))
 model.eval()
 wav2vec2_model.eval()

         return x
 model = CNN(5)
+model.load_state_dict(torch.load("best_model_CNN_bs32_lr0.001_epoch14_acc0.9244.pth", map_location=torch.device("cpu")))
 model.eval()
 wav2vec2_model.eval()