Update README.md

README.md

@@ -40,7 +40,7 @@ from transformers.utils.hub import cached_file
 from pruna_engine.PrunaModel import PrunaModel  # Step (1): install and import `pruna-engine` package.
 
 ...
-model_path = cached_file("PrunaAI/REPO", "model")  # Step (2): download the model files at `model_path`.
+model_path = cached_file("PrunaAI/stabilityai-stable-diffusion-2-1-base-smashed", "model")  # Step (2): download the model files at `model_path`.
 smashed_model = PrunaModel.load_model(model_path)  # Step (3): load the model.
 y = smashed_model(prompt="a photo of an astronaut riding a horse on mars", image_height=768, image_width=768)[0]  # Step (4): run the model.
 ```