Spaces:
Runtime error
Runtime error
# Import necessary libraries | |
import gradio as gr | |
import torchaudio | |
from audioseal import AudioSeal | |
import torch | |
# Function to detect watermarks in the audio and determine if it's AI-generated | |
def detect_watermark(file_info): | |
# Load the audio file from the uploaded file | |
audio, sr = torchaudio.load(file_info) | |
audio = audio.unsqueeze(0) # Add batch dimension as expected by the AudioSeal detector | |
# Initialize the AudioSeal detector | |
detector = AudioSeal.load_detector("audioseal_detector_16bits") | |
# Detect watermarks in the audio | |
result, message = detector.detect_watermark(audio, message_threshold=0.5) | |
# Interpret the detection result | |
detection_result = "AI-generated" if result else "genuine" | |
return f"This audio is likely {detection_result} based on watermark detection." | |
# Define Gradio interface | |
interface = gr.Interface(fn=detect_watermark, | |
inputs=gr.Audio(source="upload", type="file", label="Upload your audio"), | |
outputs="text", | |
title="Deep Fake Defender: AI Voice Cloning Detection", | |
description="Upload an audio file to check if it's AI-generated or genuine.") | |
# Launch the Gradio app | |
if __name__ == "__main__": | |
interface.launch() | |