Matej commited on
Commit
5aab43b
1 Parent(s): 096ab10
Files changed (1) hide show
  1. my_app.py +2 -0
my_app.py CHANGED
@@ -32,7 +32,9 @@ def predict(model_selection, image):
32
  model = model1 if model_selection == "EfficentNetB0 Fine Tune" else model2
33
 
34
  image = preprocess(image)
 
35
  prediction = model.predict(image)
 
36
  predicted_class = classes[int(tf.argmax(prediction, axis=1))]
37
  confidence = tf.reduce_max(prediction).numpy()
38
  return predicted_class, confidence
 
32
  model = model1 if model_selection == "EfficentNetB0 Fine Tune" else model2
33
 
34
  image = preprocess(image)
35
+ print("image", image)
36
  prediction = model.predict(image)
37
+ print("model prediction", prediction)
38
  predicted_class = classes[int(tf.argmax(prediction, axis=1))]
39
  confidence = tf.reduce_max(prediction).numpy()
40
  return predicted_class, confidence