Update app.py

app.py

@@ -38,12 +38,15 @@ def classify_audio(audio_file):
     
     # Si tu modelo necesita 3 dimensiones, redimensiona a (1, 40, 1)
     # features = features.reshape(1, 40, 1)
-    
-    # Realiza la predicción
-    model = tf.keras.models.load_model('my_model.h5')  # Carga del modelo dentro de la función
-    prediction = model.predict(features)
-    predicted_label_index = np.argmax(prediction) 
 
+    # Carga del modelo para usar la CPU
+    model = tf.keras.models.load_model('my_model.h5', compile=False)
+
+    # Realiza la predicción en la CPU
+    with tf.device('/CPU:0'):
+        prediction = model.predict(features)
+        predicted_label_index = np.argmax(prediction) 
+    
     # Devuelve la etiqueta predicha
     predicted_label = labels[predicted_label_index]
     return predicted_label