diff --git a/views/models/edit.html b/views/models/edit.html
index 0ca6ba4..4eb09b9 100644
--- a/views/models/edit.html
+++ b/views/models/edit.html
@@ -336,15 +336,6 @@
 
 {{ define "run-model-card" }}
     <form hx-headers='{"REQUEST-TYPE": "html"}' enctype="multipart/form-data" hx-post="/models/run" hx-swap="outerHTML">
-        {{ if .NotFound }}
-            <div>
-                The class was not found
-            </div>
-        {{ else if .Result }}
-            <div>
-                Img Class: {{.Result}}
-            </div>
-        {{ end }}
         <input type="hidden" name="id" value={{.Model.Id}} />
         <fieldset class="file-upload" >
             <label for="file">Image</label>
@@ -369,6 +360,20 @@
         <button>
             Run
         </button>
+        {{ if .NotFound }}
+            <div class="result">
+                <h1>
+                    The class was not found
+                </h1>
+            </div>
+        {{ else if .Result }}
+            <div>
+                <h1>
+                    Result
+                </h1>
+                The image was classified as {{.Result}}
+            </div>
+        {{ end }}
     </form>
 {{ end }}