Update app.py

app.py

@@ -1,14 +1,9 @@
-# First, install necessary packages directly in your Python environment or Colab notebook
-!pip install gradio audioseal torchaudio soundfile matplotlib
-
 import gradio as gr
 import torch
 import torchaudio
-import os
-from audioseal import AudioSeal, attacks
+from audioseal import AudioSeal
 from io import BytesIO
 
-# Function to watermark audio
 def watermark_audio(file_info):
     audio, sr = torchaudio.load(file_info)
     model = AudioSeal.load_generator("audioseal_wm_16bits")
@@ -19,7 +14,6 @@ def watermark_audio(file_info):
     output.seek(0)
     return output, 'audio/wav'
 
-# Function to detect watermark in audio
 def detect_watermark(file_info):
     audio, sr = torchaudio.load(file_info)
     audios = audio.unsqueeze(0)  # Add batch dimension
@@ -27,7 +21,6 @@ def detect_watermark(file_info):
     result, message = detector.detect_watermark(audios, message_threshold=0.5)
     return f"Watermark Detected: {result}, Message: {message.numpy()}"
 
-# Gradio interface
 with gr.Blocks() as demo:
     with gr.Row():
         with gr.Column():
@@ -44,3 +37,4 @@ with gr.Blocks() as demo:
             detect_btn.click(detect_watermark, inputs=audio_input_detect, outputs=detection_output)
 
 demo.launch()
+