Update app.py

app.py

@@ -90,6 +90,7 @@ if __name__ == "__main__":
     len_start_bit = 16
     device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
     # model = wavmark.load_model().to(device)
+    model = AudioSeal.load_generator("audioseal_wm_16bits")
     main()
 
     # audio_path = "/Users/my/Library/Mobile Documents/com~apple~CloudDocs/CODE/PycharmProjects/4_语音水印/419_huggingface水印/WavMark/example.wav"