c-walls commited on
Commit
4177ee1
·
1 Parent(s): e40bc6d

troubleshoot prediction..

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -3,7 +3,8 @@ import tensorflow as tf
3
  import numpy as np
4
 
5
  model = tf.saved_model.load("/home/user/app/model")
6
- print(dir(model))
 
7
 
8
  def mnist_classifier(img):
9
  img_tensor = tf.convert_to_tensor(img['composite'], dtype=tf.float32)
 
3
  import numpy as np
4
 
5
  model = tf.saved_model.load("/home/user/app/model")
6
+ concrete_functions = list(model.signatures.values())
7
+ print(f"Found {len(concrete_functions)} concrete functions.")
8
 
9
  def mnist_classifier(img):
10
  img_tensor = tf.convert_to_tensor(img['composite'], dtype=tf.float32)