Update app.py

app.py

@@ -58,6 +58,10 @@ def main():
         with open(tmp_input_audio_file, "wb") as f:
             f.write(audio_file.getbuffer())
 
+        wav, sample_rate = torchaudio.load(tmp_input_audio_file)
+
+        st.markdown("SR")
+        st.markdown(sample_rate)
         # 展示文件到页面上
         # st.audio(tmp_input_audio_file, format="audio/wav")
 
@@ -69,7 +73,7 @@ def main():
             if add_watermark_button:  # 点击按钮后执行的
                 if audio_file and watermark_text:
                     with st.spinner("Adding Watermark..."):
-                        watermark = model.get_watermark(tmp_input_audio_file, default_sr)
+                        watermark = model.get_watermark(wav, default_sr)
                         # watermarked_audio, encode_time_cost = add_watermark(tmp_input_audio_file, watermark_text)
 #                         st.write("Watermarked Audio:")
 #                         print("watermarked_audio:", watermarked_audio)