Update app.py

app.py

@@ -1,22 +1,19 @@
 import gradio as gr
 from transformers import pipeline
 
-# Load the deepfake audio detection model
+# Load audio classification pipeline with an open model
 model = pipeline("audio-classification", model="motheecreator/Deepfake-audio-detection")
 
-def predict(audio):
-    result = model(audio)
-    # Assuming the model returns a list of dictionaries with 'label' and 'score'
-    label = result[0]['label']
-    score = result[0]['score']
-    return f"Prediction: {label} (Confidence: {score:.2f})"
+def detect_deepfake(audio):
+    result = model(audio)[0]
+    label = result['label']
+    score = result['score']
+    return f"{label} ({round(score * 100, 2)}%)"
 
-interface = gr.Interface(
-    fn=predict,
-    inputs=gr.Audio(type="filepath"),
+gr.Interface(
+    fn=detect_deepfake,
+    inputs=gr.Audio(type="filepath", label="Upload a voice clip"),
     outputs="text",
-    title="Deepfake Audio Detector",
-    description="Upload an audio file to check if it's AI-generated or human."
-)
-
-interface.launch()
+    title="AI Voice Detector",
+    description="This tool uses a trained AI model to detect whether a voice clip is real or AI-generated."
+).launch()