Update app.py

app.py

@@ -50,14 +50,22 @@ def predict_voice(audio_file_path):
         
         logits = (outputs_original.logits + outputs_augmented.logits) / 2
         predicted_index = logits.argmax()
-        label = model.config.id2label[predicted_index.item()]
+        original_label = model.config.id2label[predicted_index.item()]
         confidence = torch.softmax(logits, dim=1).max().item() * 100
         
+        # Map original labels to new labels
+        label_mapping = {
+            "Spoof": "AI-generated Clone",
+            "Bonafide": "Real Human Voice"
+        }
+        # Use the original label to get the new label
+        new_label = label_mapping.get(original_label, "Unknown")  # Default to "Unknown" if label not found
+        
         waveform_plot = plot_waveform(waveform, sample_rate)
         spectrogram_plot = plot_spectrogram(waveform, sample_rate)
         
         return (
-            f"The voice is classified as '{label}' with a confidence of {confidence:.2f}%.",
+            f"The voice is classified as '{new_label}' with a confidence of {confidence:.2f}%.",
             waveform_plot,
             spectrogram_plot
         )