Update app.py

app.py

@@ -141,14 +141,14 @@ def predict_hate_speech(audio_path):
         predicted_class = torch.argmax(probabilities, dim=1).item()
         confidence = probabilities[0][predicted_class].item()
 
-    if confidence > 65 and predicted_class == 1:
+    if confidence > 0.65 and predicted_class == 1:
             result = {
             'Classification': 'Hate Speech\n',
             'Confidence': f"{confidence:.2%}"
         }
     else:
-        if confidence < 50 and predicted_class == 1:
-            confidence = 100 - confidence
+        if confidence < 0.5 and predicted_class == 1:
+            confidence = 1 - confidence
         result = {
             'Classification': 'Non-Hate Speech\n',
             'Confidence': f"{confidence:.2%}"