Model page tweak

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>
 

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,})