[UPD] app.py

app.py

@@ -7,7 +7,7 @@ from sklearn.preprocessing import LabelEncoder
 from tensorflow.keras.models import load_model
 
 
-le = LabelEncoder()
+# le = LabelEncoder()
 
 model = load_model('model/audio_prediction_model')
 
@@ -20,9 +20,9 @@ def prediction_audio(audiofile, model):
     X = mel_frequency.reshape(1, -1) # (1, 128)
 
     predicted_label = np.argmax(model.predict(X), axis=-1)
-    predicted_class = le.inverse_transform(predicted_label)
+    # predicted_class = le.inverse_transform(predicted_label)
 
-    return predicted_class[0]
+    return predicted_label[0]
 
 st.title("Détection de coups de feu")