diff --git a/src/mednet/libs/segmentation/scripts/predict.py b/src/mednet/libs/segmentation/scripts/predict.py index db43e0e1172e66fb05a72d05c34c711318d2c43a..b15d7dafc74becbe38ff8a0b098d6858fdc62846 100644 --- a/src/mednet/libs/segmentation/scripts/predict.py +++ b/src/mednet/libs/segmentation/scripts/predict.py @@ -66,5 +66,6 @@ def predict( json_predictions = run(model, datamodule, device_manager, output_folder) - save_json_with_backup(output_folder / "predictions.json", json_predictions) - logger.info(f"Predictions saved to `{str(json_predictions)}`") + predictions_file = output_folder / "predictions.json" + save_json_with_backup(predictions_file, json_predictions) + logger.info(f"Predictions saved to `{str(predictions_file)}`")