troubleshoot prediction..

app.py

@@ -3,8 +3,7 @@ import tensorflow as tf
 import numpy as np
 
 model = tf.saved_model.load("/home/user/app/model")
-concrete_functions = list(model.signatures.values())
-print(f"Found {len(concrete_functions)} concrete functions.")
+labels = ["zero", "one", "two", "three", "four", "five", "six", "seven", "eight", "nine"]
 
 def mnist_classifier(img):
     img_tensor = tf.convert_to_tensor(img['composite'], dtype=tf.float32)
@@ -14,9 +13,9 @@ def mnist_classifier(img):
     img_tensor /= 255.0
     img_tensor = tf.expand_dims(img_tensor, 0)
 
-    #prediction = model.signatures['serving_default'](img_tensor)
-    #print(model.signatures['serving_default'].structured_outputs)
-    #return str(tf.argmax(prediction['output_0'], axis=1).numpy()[0])
+    prediction = model.signatures['serving_default'](img_tensor)
+    print(model.signatures['serving_default'].structured_outputs)
+    return str(tf.argmax(prediction['output_0'], axis=1).numpy()[0])
 
 demo = gr.Interface(fn=mnist_classifier, inputs="sketchpad", outputs="text", title="MNIST Checker", description="Draw a number 0-9 to see if the model can classify it.")
 demo.launch()

model/saved_model.pb

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:f32d7b168ea0e701b5db839fb47ac1f7127e29e7192a30382b6b518f671864f2
-size 8715

model/variables/variables.data-00000-of-00001

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:7d99aa0a3a8fd1def733e9e015b8fe381ee09b9f379d48a5d7bca43733d4ccde
-size 1075500