Update app.py

app.py

@@ -3,7 +3,6 @@ import numpy as np
 import tensorflow as tf
 import librosa
 import librosa.util
-import pickle
 from sklearn.preprocessing import LabelEncoder
 
 # Feature extraction function
@@ -43,9 +42,20 @@ def predict_class(file_path, model, label_encoder):
 # Load the pre-trained model
 model = tf.keras.models.load_model("voice_classification_modelm.h5")
 
-# Load the label encoder
-with open("label_encoder.pkl", "rb") as f:
-    label_encoder = pickle.load(f)
+# Define the class labels (same as used during training)
+class_labels = [
+    "all_vowels_healthy",
+    "allvowels_functional",
+    "allvowels_laryngitis",
+    "allvowels_lukoplakia",
+    "allvowels_psychogenic",
+    "allvowels_rlnp",
+    "allvowels_sd"
+]
+
+# Initialize the LabelEncoder
+label_encoder = LabelEncoder()
+label_encoder.fit(class_labels)
 
 # Define the Gradio function
 def classify_audio(audio_file):