Update app.py
Browse files
app.py
CHANGED
@@ -54,7 +54,7 @@ def predict(img) -> Tuple[Dict,float] :
|
|
54 |
|
55 |
pred_img = model.predict(tf.expand_dims(image, axis=0))
|
56 |
pred_class = class_names[pred_img.argmax()]
|
57 |
-
|
58 |
|
59 |
|
60 |
|
|
|
54 |
|
55 |
pred_img = model.predict(tf.expand_dims(image, axis=0))
|
56 |
pred_class = class_names[pred_img.argmax()]
|
57 |
+
print(f"Predicted brain tumor is: {pred_class} with probability: {pred_img.max():.2f}")
|
58 |
|
59 |
|
60 |
|