fix build error

app.py

@@ -1,4 +1,5 @@
 import time
+import wavmark
 import streamlit as st
 import os
 import torch
@@ -6,6 +7,10 @@ import datetime
 import numpy as np
 import soundfile
 from wavmark.utils import file_reader
+import subprocess
+import sys
+import time
+
 
 
 def my_read_file(audio_path, max_second):
@@ -35,17 +40,18 @@ def add_watermark(audio_path, watermark_text):
 def decode_watermark(audio_path):
     assert os.path.exists(audio_path)
 
-    
+    #t1 = time.time()
     signal, sr, audio_length_second = my_read_file(audio_path, max_second_decode)
     payload_decoded, _ = wavmark.decode_watermark(model, signal, show_progress=False)
-    decode_cost = time.time() - t1
+    #decode_cost = time.time() - t1
 
     if payload_decoded is None:
-        return "No Watermark", decode_cost
+       #return "No Watermark" , decode_cost
+       return "No Watermark"
 
     payload_decoded_str = "".join([str(i) for i in payload_decoded])
     st.write("Result:", payload_decoded_str)
-    
+    #st.write("Time Cost:%d seconds" % (decode_cost))
 
 
 def create_default_value():
@@ -59,7 +65,7 @@ def create_default_value():
 def main():
     create_default_value()
 
-    # st.title("AudioWaterMarking")
+    st.title("AudioWaterMarking")
     markdown_text = """
     # Audio WaterMarking
     You can upload an audio file and encode a custom 16-bit watermark or perform decoding from a watermarked audio.

requirements.txt

@@ -1,3 +1,4 @@
+git+https://github.com/wavmark/wavmark
 audioread==3.0.1
 certifi==2023.11.17
 cffi==1.16.0
@@ -32,14 +33,14 @@ resampy==0.4.2
 scikit-learn==1.4.0
 scipy==1.12.0
 simpleaudio==1.0.4
-soundfile==0.12.1
+soundfile
 soxr==0.3.7
 sympy==1.12
 threadpoolctl==3.2.0
-torch
+torch==2.1.2
 torchaudio==2.1.2
 tqdm==4.66.1
 typing_extensions==4.9.0
 urllib3==2.1.0
-wavmark
+wavmark==0.0.3