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/segmind-SSD-1B-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=1024, image_width=1024)[0]  # Step (4): run the model.
 ```