Update app.py

app.py

@@ -205,12 +205,13 @@ def main():
                 
             if detect_watermark_button:
                 with st.spinner("Detecting..."):
-                    # result, message = detector.detect_watermark(watermarked_audio, sample_rate=default_sr, message_threshold=0.5)
-                    # st.markdown("This is likely a watermarked audio:")
-                    # st.markdown(result)
-                    # print(f"\nThis is likely a watermarked audio: {result}")
+                    result, message = detector.detect_watermark(watermarked_audio, sample_rate=default_sr, message_threshold=0.5)
+                    st.markdown("Probability of audio being watermarked: ")
+                    st.markdown(result)
+                    st.markdown("This is likely a watermarked audio: ")
+                    print(f"\nThis is likely a watermarked audio: {result}")
 
-                    # Run on an unwatermarked audio
+                    Run on an unwatermarked audio
 
                     if file_extension in [".wav", ".flac"]: 
                         wav, sample_rate = torchaudio.load("test.wav")
@@ -225,9 +226,11 @@ def main():
                         wav= wav.unsqueeze(0)
                         
                     result2, message2 = detector.detect_watermark(wav, sample_rate=default_sr, message_threshold=0.5)
-                    st.markdown("This is likely an unwatermarked audio:")
                     print(f"This is likely an unwatermarked audio: {result2}")
+                    st.markdown("Probability of audio being watermarked (Original Audio): ")
+                    st.markdown(result)
                     st.markdown(result2)
+                    st.markdown("This is likely an unwatermarked audio:")
 
 
 if __name__ == "__main__":