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 |
|