Update README.md

README.md

@@ -40,10 +40,13 @@ As of now to use the model you have to install the [PerceptNet repo](https://git
 
 ```python
 from perceptnet.networks import PerceptNet
+from tensorflow.keras.utils import get_file
 
 weights_path = get_file(fname='perceptnet_rgb.h5',
-                        origin='https://huggingface.co/Jorgvt/PerceptNet/blob/main/final_model_rgb.h5')
+                        origin='https://huggingface.co/Jorgvt/PerceptNet/resolve/main/tf_model.h5')
 model = PerceptNet(kernel_initializer='ones', gdn_kernel_size=1, learnable_undersampling=False)
 model.build(input_shape=(None, 384, 512, 3))
 model.load_weights(weights_path)                        
-```
+```
+
+> PerceptNet requires `wandb` to be installed. It's something we're looking into.