fix main py

app/main.py

@@ -229,9 +229,13 @@ def extract_features(file_path):
     # features = pd.concat([mfccs_mean, spectral_centroids, spectral_rolloff, spectral_flux, spectral_contrast, pitch_mean, zero_crossings])
     
     features = smile.process_file(file_path)
-    all_data = pd.concat([features], ignore_index=True)
+    features_reshaped = features.squeeze()
 
-    return pd.DataFrame([all_data])
+    # Ensure it's now a 2D structure suitable for DataFrame
+    print("New shape of features:", features_reshaped.shape)
+
+    all_data = pd.DataFrame([features_reshaped])
+    return all_data
 
 
 @app.post("/mlp")
@@ -264,7 +268,7 @@ async def handle_audio(file: UploadFile = File(...)):
         features = mlp_pca.transform(features)
 
         # Dummy example to proceed with an inference
-        results = mlp_model.predict([features])
+        results = mlp_model.predict(features)
 
         # Clean up (optional, especially if dealing with large files or sensitive data)
         os.remove(temp_filename)