Update app.py

app.py

@@ -118,28 +118,24 @@ def process_audio(audio1, audio2):
         probabilities = torch.softmax(output, dim=1)[0]
         confidence = probabilities[prediction].item()
         
+        label = "Original" if prediction == 0 else "Deepfake/Impersonation"
+        
         result = f"""
         📊 Authentication Results:
         
         🔹 Similarity Score: {score:.4f}
-        🔹 Classification: {model_config['labels'][prediction]}
+        🔹 Classification: {label}
         🔹 Confidence: {confidence:.4f}
         
-        # Analysis Details:
-        # - A similarity score closer to 1.0 indicates higher voice similarity
-        # - Classification indicates the model's assessment of the voice comparison
-        # - Confidence shows how certain the model is about its classification
-        
         {'⚠️ Potential Voice Impersonation Detected!' if prediction > 0 else '✅ Authentic Voice Match'}
         """
-        
+
         return result
     
     except Exception as e:
         return f"An error occurred: {str(e)}"
     
     finally:
-        # Clean up temporary files
         if os.path.exists(temp_path1):
             os.remove(temp_path1)
         if os.path.exists(temp_path2):