Update README.md

README.md

@@ -25,17 +25,17 @@ The US's manufacturing and national security sectors heavily rely on a diverse r
 
 | File Name | Description |
 |-----------|-------------|
-| `GAN_model.py` | V-Unet and GAN model script |
+| `GAN_model.py` | Training GAN model for POC |
 | `README.md` | Documentation of the project |
-| `VAE-unet.py` | VAE-Unet model script |
+| `VAE-unet.py` | Create and train VAE U-net model for map segmentation |
+| `VAE-unet_updated.py` | Create and train VAE U-net model for map segmentation with updated loss function |
 | `create_prediction_map.py` | Data loader class script for input processing |
-| `data_util.py` | Utility functions for V-Unet and GAN model |
+| `data_util.py` | Utility functions for dataloader that read and process data from TIFF and JSON files |
 | `eval_gan.py` | Evaluation script for GAN model |
 | `eval_vauner.py` | Evaluation script for VAE-Unet model |
-| `train_model.py` | Training script for the model |
-| `train_model_u_legends.py` | Training script with legends |
-| `two-h-model.py` | Script for V-Unet and GAN model |
-| `unet_util.py` | Utility functions for the Unet model |
+| `train_model.py` | Training script for the attention model with processed legends|
+| `train_model_u_legends.py` | Training script for the attention model with unprocess legends|
+| `unet_util.py` | Utility functions for the attention U-net model |
 
 For more information on specific files or collaboration, please refer to the contribution guidelines or reach out to the repository maintainers.