Update README.md

README.md

@@ -33,15 +33,12 @@ from huggingface_hub import snapshot_download
 
 ### Download the repo files
 ```python
-hf_dir = snapshot_download(
-    repo_id="alexanderkroner/MSI-Net", allow_patterns=["*.pb", "*.jpg"]
-)
+hf_dir = snapshot_download(repo_id="alexanderkroner/MSI-Net")
 ```
 
 ### Load the saliency model
 ```python
-model = tf.saved_model.load(hf_dir)
-model = model.signatures["serving_default"]
+model = tf.keras.models.load_model(hf_dir)
 ```
 
 ### Load the functions for preprocessing the input and postprocessing the output