tolist()

app.py

@@ -261,7 +261,7 @@ def predict_malware(img_array: np.ndarray) -> str:
         prediction = app.state.model.predict(img_array)
         return {
             "result": MAL_CLASSES[np.argmax(prediction)],
-            "all_results": dict(zip(MAL_CLASSES, softmax(prediction)))
+            "all_results": dict(zip(MAL_CLASSES, softmax(prediction).tolist()))
         }
     except Exception as e:
         logger.error(f"Prediction error: {str(e)}")