New models

PR conflict resolution plus making HF upload more generic

fix visualization stage.

committing after a bunch of fixes and before re-running with dvc

Applied fixes to dagshub logger

fixes

fixes