Update app.py

app.py

@@ -202,13 +202,13 @@ def main():
                 size = watermarked_audio.size()
                 #st.markdown(size)
 
-                print(watermarked_audio.squeeze())
+                #print(watermarked_audio.squeeze())
                 # squeeze = watermarked_audio.squeeze(1)
                 # shape = squeeze.size()
                 
             if detect_watermark_button:
                 with st.spinner("Detecting..."):
-                    result, message = detector.detect_watermark(squeeze, sample_rate=default_sr, message_threshold=0.5)
+                    result, message = detector.detect_watermark(watermarked_audio, sample_rate=default_sr, message_threshold=0.5)
                     st.markdown(result)
                     print(f"\nThis is likely a watermarked audio: {result}")