Apply suggestions from code review

Author: Borda

examples/train-resume.py

@@ -8,7 +8,7 @@
     dataset = tv.datasets.MNIST(".", download=True, transform=tv.transforms.ToTensor())
     train, val = data.random_split(dataset, [55000, 5000])
 
-    model_path = download_model(name="jirka/kaggle/lit-auto-encoder-simple:latest", download_dir="my_models")
+    model_path = download_model(name="jirka/kaggle/lit-auto-encoder-simple", download_dir="my_models")
     print(f"model: {model_path}")
     # autoencoder = LitAutoEncoder.load_from_checkpoint(checkpoint_path=model_path)