diff --git a/prediction_service/__pycache__/views.cpython-310.pyc b/prediction_service/__pycache__/views.cpython-310.pyc
index 0dc6061a5ae930dd20530bcc4a1593620eaa8778..96afc8e415e559fd9782d811855bdfb8eb3151bf 100644
Binary files a/prediction_service/__pycache__/views.cpython-310.pyc and b/prediction_service/__pycache__/views.cpython-310.pyc differ
diff --git a/prediction_service/templates/prediction_service/model.html b/prediction_service/templates/prediction_service/model.html
index 8aabd3efb3e6a2e713ed9ea8ed267d664d9d490c..9303fdad3d3358b8bf0697cbc4a54907a5e71ecf 100644
--- a/prediction_service/templates/prediction_service/model.html
+++ b/prediction_service/templates/prediction_service/model.html
@@ -9,7 +9,7 @@
 
     <img src="data:image/png;base64,{{ img_str }}" alt="PIL Image">
 
-    <h1>{{ layer }}</h1>
+    <h2>Model Layers: {{ layer }}</h2>
     
 </div>
 
diff --git a/prediction_service/views.py b/prediction_service/views.py
index d6cf0aac8f118a384bcf26f94401fbb908790a89..8883186b774deabe8a00da7f96741809301b7f23 100644
--- a/prediction_service/views.py
+++ b/prediction_service/views.py
@@ -57,9 +57,6 @@ import base64
 @login_required
 def mlmodel(request, pk):
 
-    mlmodel = MLModel.objects.get(id=pk)
-    layercount = str(mlmodel.layers)
-
     matrix = np.asarray([[0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0],
                      [0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0],
                      [0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0],
@@ -103,5 +100,8 @@ def mlmodel(request, pk):
     img.save(buffered, format='PNG')
     img_str = base64.b64encode(buffered.getvalue()).decode('utf-8')
 
+    mlmodel = MLModel.objects.get(id=pk)
+    layercount = str(mlmodel.layers)
+
     return render(request, "prediction_service/model.html", {'img_str': img_str,
                                                              'layer': layercount,})