Spaces:
Sleeping
Sleeping
Matej
commited on
Commit
•
5aab43b
1
Parent(s):
096ab10
debug
Browse files
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
|