updated readme and pytorch version

README.md

@@ -120,7 +120,8 @@ python scripts/pla_net_inference.py \
 
 ## Models
 
-You can download the pre-trained models from [Hugging Face](https://huggingface.co/juliocesar-io/PLA-Net).
+You can download the pre-trained models from the [Hugging Face Model Card](https://huggingface.co/juliocesar-io/PLA-Net/tree/main/checkpoints).
+
 ## Training 
 
 To train each of the components of our method: LM, LM+Advs, LMPM and PLA-Net please refer to planet.sh file and run the desired models.

gradio/title.md

@@ -6,7 +6,7 @@
     <br>
     <div style="display: flex; justify-content: center; align-items: center; text-align: center;">
         <a href="#"><img src="https://img.shields.io/static/v1?label=11.6&message=CUDA&color=green"></a> &ensp;
-        <a href="#"><img src="https://img.shields.io/static/v1?label=1.12.0&message=Pytorch&color=red"></a> &ensp;
+        <a href="#"><img src="https://img.shields.io/static/v1?label=1.13.1&message=Pytorch&color=red"></a> &ensp;
         <a href="https://github.com/juliocesar-io/PLA-Net"><img src="https://img.shields.io/static/v1?logo=github&label=Github&message=Fork"></a>
     </div>
     <br>