Update

app.py

@@ -6,27 +6,33 @@ import traceback
 
 def detect_watermark(audio_file_path):
     try:
-        # Load the audio file using torchaudio
+        # Load the audio file
         waveform, sample_rate = torchaudio.load(audio_file_path)
 
-        # Ensure waveform is 2D (channels, samples). If it's mono, add an axis.
-        if waveform.ndim == 2:
-            waveform = waveform.unsqueeze(0)  # Add batch dimension
+        # Ensure waveform has a batch dimension for processing
+        if waveform.ndim < 3:
+            waveform = waveform.unsqueeze(0)
 
-        # Initialize and use the AudioSeal detector
+        # Initialize the AudioSeal detector
         detector = AudioSeal.load_detector("audioseal_detector_16bits")
-        result, _ = detector.detect_watermark(waveform, message_threshold=1)
+        
+        # Set a conservative threshold for watermark detection
+        message_threshold = 0.7  # A higher threshold means more confidence is required to classify as AI-generated
+        
+        result, confidence = detector.detect_watermark(waveform, message_threshold=message_threshold)
 
-        # Interpret and return the detection result
-        detection_result = "AI-generated" if result else "genuine"
-        return f"This audio is likely {detection_result} based on watermark detection."
+        # Interpret the detection result
+        if result:
+            detection_result = f"AI-generated with confidence {confidence}"
+        else:
+            detection_result = "Genuine or the AI watermark is undetectable at the current threshold"
 
+        return f"This audio is likely {detection_result}."
     except Exception as e:
-        # Log the error with traceback for debugging
         error_traceback = traceback.format_exc()
         return f"Error occurred: {e}\n\n{error_traceback}"
 
-# Define Gradio interface with "filepath" type for audio input
+# Define the Gradio interface
 interface = gr.Interface(
     fn=detect_watermark,
     inputs=gr.Audio(label="Upload your audio", type="filepath"),