Update index.html

index.html

@@ -791,20 +791,35 @@ <h5 class="sub-titles">Training the beast</h5>
             <p class="main-sub-text">Training a standard model is rather simple: Get the data, get the model, feed the model the data :). Yet, not all is happiness, neural networks require more work. Usually we will have 
                 to feed them the data more than ones (have some epochs). Therefore, we can see how it evolves over time.
                 <br><br>
-
+            </p>
+            <div class="image-section">
+                            <img src="{{ site.baseurl }}assets/images/graphs/models/Dense-shape.png" alt="Dense-shape">
+                        </div>
+                        <div class="image-section">
+                            <img src="{{ site.baseurl }}assets/images/graphs/models/U-shape.png" alt="U-shape">
+                        </div>
+            <p class="main-sub-text">
                 Unlikely, our big boys are not very strong, and they had a bad time training :(. The lack of computational power does not allow us to make more validation on this models, and therefore train longer and with better settings. We leave this as a learning taks for the reader. So now, let's compare the performance of the models.
                 <br><br>
-
-
+            </p>
+                <div class="container">
+                        <div class="image-section">
+                            <img src="{{ site.baseurl }}assets/images/graphs/models/score_comparation.png" alt="Score comparison">
+                        </div>
+                </div>
+
+<p class="main-sub-text">
                 As we can see, all of them have a high accuracy and decent f-score. We also tried doing the predictions at random and as we can see, our models are better! Now, let’s see some of the predictions. 
                 <br><br>
 
 
                 As we can see our ridge guy get some a lot of the genres right but then, add some other for you to be aware of any potential topics hahaha!
 
+            </p>
+
+
 
 
-            </p>
 
 
 </main>