Update README.md

README.md

@@ -10,11 +10,20 @@ pipeline_tag: text-to-3d
 # Datasets and models for Shap-MeD
 Please refer to the following links to understand how to use it:
 - [Shap-MeD Article](https://arxiv.org/abs/2503.15562): Theoretical foundation of the project.
-- [Shap-MeD repository](https://github.com/jd-rodriguezp1234/Shap-MeD/tree/main/dataset): This repository contains the code and instructions you want to use these files directly.
+- [Shap-MeD repository](https://github.com/jd-rodriguezp1234/Shap-MeD/): This repository contains the code and instructions you want to use these files directly.
 
 ## Abstract
 We present Shap-MeD, a text-to-3D object generative model specialized in the biomedical domain. The objective of this study is to develop an assistant that facilitates the 3D modeling of medical objects, thereby reducing development time. 3D modeling in medicine has various applications, including surgical procedure simulation and planning, the design of personalized prosthetic implants, medical education, the creation of anatomical models, and the development of research prototypes. To achieve this, we leverage Shap-e, an open-source text-to-3D generative model developed by OpenAI, and fine-tune it using a dataset of biomedical objects. Our model achieved a mean squared error (MSE) of 0.089 in latent generation on the evaluation set, compared to Shap-e's MSE of 0.147. Additionally, we conducted a qualitative evaluation, comparing our model with others in the generation of biomedical objects. Our results indicate that Shap-MeD demonstrates higher structural accuracy in biomedical object generation
 
+## Files
+These are the dataset and model files in this repository. However it is heavily encouraged to read the [Shap-MeD repository](https://github.com/jd-rodriguezp1234/Shap-MeD/) README to use them. 
+
+### shap-med-dataset.tar.gz
+This file contains the latent **.pt** files required to train the Shap-e model.
+
+### shap-med-models.tar.gz
+This file contains the trained **.pt** models required to generate meshes from text. 
+
 ## How to cite
 Please cite by using the next Bibtex citation:
 ```