Update app.py
Browse files
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"):
|