Update README.md

README.md

@@ -4,7 +4,7 @@ library_name: keras
 
 ## Model description
 
-This UNet model identifies contrails in satellite images. It takes pre-processed .npy files (images) from the OpenContrails dataset [here](https://arxiv.org/abs/2304.02122) as its inputs, and returns a "mask" image showing only the contrails overlayed on the same area. 
+This TransUNet model identifies contrails in satellite images. It takes pre-processed .npy files (images) from the OpenContrails dataset [here](https://arxiv.org/abs/2304.02122) as its inputs, and returns a "mask" image showing only the contrails overlayed on the same area. 
 We achieve a Mean IOU of 0.6951 on the validation set.
 
 ## Intended uses