Fix ML pipeline (ML)

[jasozh]

Branch fix-ml-pipeline

Task at Hand

• The entire pipeline from ml.main is currently working. However, the output result is really bad for some reason. Debug and fix the issue.
• Add difficulty levels to the model in the form of easy, medium, and hard corresponding to the level of detail. This should affect either the TEED hyperparams or the edge smoothing dilation params.