Update app.py
Browse files
app.py
CHANGED
@@ -234,7 +234,7 @@ def main():
|
|
234 |
# Export it as a WAV file
|
235 |
audio.export("test.wav", format="wav")
|
236 |
wav3, sample_rate = torchaudio.load("test.wav")
|
237 |
-
|
238 |
|
239 |
result2, message2 = detector.detect_watermark(wav3, sample_rate=default_sr, message_threshold=0.5)
|
240 |
print(f"This is likely an unwatermarked audio: {result2}")
|
|
|
234 |
# Export it as a WAV file
|
235 |
audio.export("test.wav", format="wav")
|
236 |
wav3, sample_rate = torchaudio.load("test.wav")
|
237 |
+
wav3= wav3.unsqueeze(0)
|
238 |
|
239 |
result2, message2 = detector.detect_watermark(wav3, sample_rate=default_sr, message_threshold=0.5)
|
240 |
print(f"This is likely an unwatermarked audio: {result2}")
|