deveix commited on
Commit
371eac7
·
1 Parent(s): 39bd3a6

fix prediction

Browse files
Files changed (1) hide show
  1. app/main.py +1 -1
app/main.py CHANGED
@@ -264,7 +264,7 @@ async def handle_audio(file: UploadFile = File(...)):
264
  features = mlp_pca.transform(features)
265
 
266
  # Dummy example to proceed with an inference
267
- results = mlp_model.predict(features)
268
 
269
  # Clean up (optional, especially if dealing with large files or sensitive data)
270
  os.remove(temp_filename)
 
264
  features = mlp_pca.transform(features)
265
 
266
  # Dummy example to proceed with an inference
267
+ results = mlp_model.predict([features])
268
 
269
  # Clean up (optional, especially if dealing with large files or sensitive data)
270
  os.remove(temp_filename)