Update hf_upload.py

Update evaluate_model.py

Update evaluate_model.py

Update evaluate_model.py

Update train_model.py

Update train_model.py

Fixes and updates

Updates

Update predict_model.py

Fixes and almost done

Update predict_model.py

Update predict_model.py

Update predict_model.py

Update predict_model.py

Update predict_model.py

Update predict_model.py

Update predict_model.py

Update predict_model.py

Update predict_model.py

Update predict_model.py

Update evaluate_model.py

Update evaluate_model.py

Update model.py

Update model.py

Update model.py

Update model.py

Update model.py

Update model.py

Update model.py

Update model.py

Update model.py

Update evaluate_model.py

Updates to training_params

Merge branch 'master' of https://dagshub.com/gagan3012/summarization

Change log parameters

Properly save the model in order to upload to the HF model hub

PR conflict resolution plus making HF upload more generic

Updates

fix visualization stage.

new updates

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

Merge commit '8d1f074f67512e839e8d290ade59fc8fe73f7c9c' into fix-mlflow

Fixing MLflow logging by using the MLflow autolog feature for Pytorch instead of the built-in pytorch lightning MLflowLogger

updates

Applied fixes to dagshub logger

Starting to apply fixes for the project to latest version

updates

updates

black style added

fixes