Update app.py

app.py

@@ -70,7 +70,7 @@ def detect_watermark(audio_data, sample_rate):
     results, messages = detector.forward(waveform.unsqueeze(0), sample_rate=sample_rate)  # Ensure waveform is in batch form
     detect_probs = results[:, 1, :]
     result = detect_probs.mean().cpu().item()
-    message = f"Detection result: {'Watermarked Audio' if result > 0.5 else 'Not watermarked'}\nProbability of watermark: {result}"
+    message = f"Detection result: {'Watermarked Audio' if result > 0.5 else 'Not watermarked'}"
     spectrogram_image = plot_spectrogram(waveform, sr)
     
     return message, spectrogram_image