Add application file

app.py

@@ -25,7 +25,7 @@ def add_watermark(audio_path, watermark_text):
     _, signal_wmd, time_cost = wm_add_v2.add_watermark(watermark, data, 16000, 0.1, device, model)
 
     tmp_file_name = datetime.datetime.now().strftime('%Y-%m-%d_%H-%M-%S') + "_" + str(uuid.uuid4()) + ".wav"
-    tmp_file_path = 'temp/' + tmp_file_name
+    tmp_file_path = '/tmp/' + tmp_file_name
     soundfile.write(tmp_file_path, signal_wmd, sr)
     return tmp_file_path
 
@@ -64,7 +64,7 @@ def main():
     if action == "Add Watermark":
         audio_file = st.file_uploader("Upload Audio File (WAV)", type=["wav"], accept_multiple_files=False)
         if audio_file:
-            tmp_input_audio_file = os.path.join("temp", audio_file.name)
+            tmp_input_audio_file = os.path.join("/tmp/", audio_file.name)
             with open(tmp_input_audio_file, "wb") as f:
                 f.write(audio_file.getbuffer())
             st.audio(tmp_input_audio_file, format="audio/wav")
@@ -94,7 +94,7 @@ def main():
         if audio_file:
             if st.button("Decode Watermark"):
                 # 1.保存
-                tmp_file_for_decode_path = os.path.join("temp", audio_file.name)
+                tmp_file_for_decode_path = os.path.join("/tmp/", audio_file.name)
                 with open(tmp_file_for_decode_path, "wb") as f:
                     f.write(audio_file.getbuffer())