From 8a24a9ec287b8626aee2d4f72aebdc439f4fae24 Mon Sep 17 00:00:00 2001
From: Nikolaos Zioulis <5476245+zokin@users.noreply.github.com>
Date: Wed, 3 Aug 2022 10:49:59 +0300
Subject: [PATCH] Update README.md

---
 README.md | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/README.md b/README.md
index 44c36aa..2caef7d 100644
--- a/README.md
+++ b/README.md
@@ -49,17 +49,20 @@ We generate a dataset (using GibsonV2) and provide baselines for holistic perfor
 - [x] Web Demo
 - [x] Data Download
 - [x] Loader & Splits
-- [ ] Models Weights Download
+- [x] Model Code & Models Weights
 - [ ] Model Serve Code
 - [ ] Model Hub Code
-- [ ] Metrics Code
+- [x] Metrics Code
 
 ### Demo
 
 A publicly hosted demo of the baseline models can be found [here](https://share.streamlit.io/tzole1155/threedit).
 Using the web app, it is possible to upload a panorama and download a 3D reconstructed mesh of the scene using the derived depth map.
+
 > Note that due to the external host's caching issues, it might be necessary to refresh your browser's cache in between runs to update the 3D models.
 
+> The model's [code](https://github.com/tzole1155/ThreeDit/blob/master/unet/model.py) and [weights](https://github.com/tzole1155/ThreeDit/releases/download/Unet/unet.pth) are also available in our demo.
+
 ### Data
 
 #### Download