Spaces:
Runtime error
Runtime error
File size: 1,190 Bytes
f4f5a40 488d50e f03ec98 dff69a4 d7a0eb1 dff69a4 d7a0eb1 1c53cc7 d7a0eb1 1c53cc7 d7a0eb1 faee536 f03ec98 6853f8f 1c53cc7 faee536 f03ec98 faee536 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 |
import gradio as gr
from audioseal import AudioSeal
import torch
import numpy as np
import traceback
def detect_watermark(audio_data, sample_rate):
try:
audio_array, _ = audio_data
if audio_array.ndim == 1:
audio_array = np.expand_dims(audio_array, axis=0)
waveform = torch.tensor(audio_array, dtype=torch.float32)
if waveform.ndim == 2:
waveform = waveform.unsqueeze(0)
detector = AudioSeal.load_detector("audioseal_detector_16bits")
result, message = detector.detect_watermark(waveform, message_threshold=0.5)
detection_result = "AI-generated" if result else "genuine"
return f"This audio is likely {detection_result} based on watermark detection."
except Exception as e:
error_traceback = traceback.format_exc()
return f"Error: {str(e)}\n{error_traceback}"
interface = gr.Interface(fn=detect_watermark,
inputs=[gr.Audio(label="Upload your audio", type="numpy"),
gr.Number(label="Sample Rate", value=44100, visible=False)],
outputs="text")
if __name__ == "__main__":
interface.launch()
|