feat: closes #23

views/models/edit.html

@@ -294,7 +294,11 @@
 
 {{ define "run-model-card" }}
     <form hx-headers='{"REQUEST-TYPE": "html"}' enctype="multipart/form-data" hx-post="/models/run" hx-swap="outerHTML">
-        {{ if .Result }}
+        {{ if .NotFound }}
+            <div>
+                The class was not found
+            </div>
+        {{ else if .Result }}
             <div>
                 Img Class: {{.Result}}
             </div>

views/py/python_model_template.py

@@ -34,7 +34,7 @@ def pathToLabel(path):
 def decode_image(img):
   # channels were reduced to 1 since image is grayscale
   # TODO chnage channel number based if grayscale
-  img = tf.io.decode_png(img, channels=1)
+  img = tf.io.decode_png(img, channels=3)
 
   return tf.image.resize(img, image_size)