Add code sample to load models to READMEs

Author: LTMeyer

the_well/benchmark/models/tfno/README.md

@@ -70,3 +70,11 @@ Below is the list of checkpoints available for the training of TFNO on different
 | [viscoelastic_instability](https://huggingface.co/polymathic-ai/TFNO-viscoelastic_instability) | 5E-3 | 199 | 0.7021 |
 
 ## Loading the model from Hugging Face
+
+To load the TFNO model trained on a dataset of the Well, use the following while replacing `<datasetname>` by the actual name of the dataset.
+
+```python
+from the_well.benchmark.models import TFNO
+
+model = TFNO.from_pretrained("polymathic-ai/TFNO-<datasetname>")
+```

the_well/benchmark/models/unet_classic/README.md

@@ -72,3 +72,11 @@ Below is the list of checkpoints available for the training of U-Net on differen
 | [viscoelastic_instability](https://huggingface.co/polymathic-ai/UNET-viscoelastic_instability) | 5E-4 | 198 | 0.3147 |
 
 ## Loading the model from Hugging Face
+
+To load the U-Net model trained on a dataset of the Well, use the following while replacing `<datasetname>` by the actual name of the dataset.
+
+```python
+from the_well.benchmark.models import UNetClassic
+
+model = UNetClassic.from_pretrained("polymathic-ai/UNET-<datasetname>")
+```

the_well/benchmark/models/unet_convnext/README.md

@@ -74,3 +74,11 @@ Below is the list of checkpoints available for the training of CNextU-Net on dif
 | [viscoelastic_instability](https://huggingface.co/polymathic-ai/CNextU-Net-viscoelastic_instability) | 5E-4 | 114 | 0.1966 |
 
 ## Loading the model from Hugging Face
+
+To load the CNextU-Net model trained on a dataset of the Well, use the following while replacing `<datasetname>` by the actual name of the dataset.
+
+```python
+from the_well.benchmark.models import UNetConvNext
+
+model = UNetConvNext.from_pretrained("polymathic-ai/CNextU-Net-<datasetname>")
+```