Spaces:

abdullahsajid
/

antispoofing-test

Sleeping

abdullahsajid commited on Sep 16, 2024

Commit

53c7fc4

verified ·

1 Parent(s): dcd0abb

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -300,7 +300,7 @@ voice_model_binary.heads=nn.Sequential(
         )
 voice_binary_model = UpdatedViT(voice_model_binary)
-voice_binary_model.load_state_dict(torch.load('binary_vit_model_correct.pth',map_location='cpu'))
 voice_binary_model.eval()

         )
 voice_binary_model = UpdatedViT(voice_model_binary)
+voice_binary_model.load_state_dict(torch.load('voice_weights.pth',map_location='cpu'))
 voice_binary_model.eval()