Update app.py

app.py

@@ -194,13 +194,13 @@ def main():
         elif action == "Detect Watermark":
             detect_watermark_button = st.button("Detect Watermark", key="detect_watermark_btn")
 
-            # if audio_file:
-            #     #1st attempt
-            #     watermark = model.get_watermark(wav, default_sr)
-            #     watermarked_audio = wav + watermark
-            #     print(watermarked_audio.size())
-            #     size = watermarked_audio.size()
-            #     #st.markdown(size)
+            if audio_file:
+                #1st attempt
+                watermark = model.get_watermark(wav, default_sr)
+                watermarked_audio = wav + watermark
+                print(watermarked_audio.size())
+                size = watermarked_audio.size()
+                #st.markdown(size)
 
                 
             if detect_watermark_button: