Zw07 commited on
Commit
7d506c6
·
verified ·
1 Parent(s): 95aafd2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -37
app.py CHANGED
@@ -108,14 +108,6 @@ def main():
108
  wav= wav.unsqueeze(0)
109
 
110
  elif file_extension == ".mp3":
111
-
112
- # Switch to FFmpeg backend
113
- torchaudio.set_audio_backend("sox_io")
114
-
115
- # Define CodecConfig for MP3 compression
116
- #codec_config = CodecConfig(format="mp3", compression=128) # 128 kbps for MP3
117
- #wav3, sample_rate = librosa.load("test.mp3")
118
-
119
  # Load an MP3 file
120
  audio = AudioSegment.from_mp3("test.mp3")
121
 
@@ -198,35 +190,6 @@ def main():
198
  mime="audio/mpeg",
199
  )
200
 
201
- # st.download_button(
202
- # label="Download Watermarked audio",
203
- # data="output.wav",
204
- # file_name="output.wav",
205
- # mime="audio/wav",
206
- # )
207
-
208
- #2nd Attempt
209
- # watermarked_audio = model(wav, sample_rate=default_sr, alpha=1)
210
- # print(watermarked_audio.size())
211
- # size = watermarked_audio.size()
212
- # st.markdown(size)
213
-
214
- # print(watermarked_audio.squeeze())
215
- # squeeze = watermarked_audio.squeeze(1)
216
- # shape = squeeze.size()
217
- # st.markdown(shape)
218
-
219
- # st.markdown(squeeze)
220
- # # watermarked_audio, encode_time_cost = add_watermark(tmp_input_audio_file, watermark_text)
221
- # st.write("Watermarked Audio:")
222
- # st.markdown(watermarked_audio)
223
- # print("watermarked_audio:", watermarked_audio)
224
-
225
- # watermarked_audio = torchaudio.save("output.wav", squeeze, default_sr)
226
- # st.audio(watermarked_audio, format="audio/wav")
227
- #st.write("Time Cost: %d seconds" % encode_time_cost)
228
-
229
- # # st.button("Add Watermark", disabled=False)
230
  elif action == "Detect Watermark":
231
  Detect_watermark_button = st.button("Detect Watermark", key="detect_watermark_btn")
232
  if st.button("Detect"):
 
108
  wav= wav.unsqueeze(0)
109
 
110
  elif file_extension == ".mp3":
 
 
 
 
 
 
 
 
111
  # Load an MP3 file
112
  audio = AudioSegment.from_mp3("test.mp3")
113
 
 
190
  mime="audio/mpeg",
191
  )
192
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
193
  elif action == "Detect Watermark":
194
  Detect_watermark_button = st.button("Detect Watermark", key="detect_watermark_btn")
195
  if st.button("Detect"):