Currently the Run Prediction UI is:

In the MC dropout result, we don't know the name of the prediction label and what are the names of the classes 0, 1, 2, …, 9
The predicted class should be the class with the highest mean in the MC dropout distribution.
Therefore, can we do the following adjustments in the current Run Prediction UI
- Display the name of the prediction result (e.g. X-ray: Positive or Negative)
- Display the names of the classes in MC dropout (in additional to the class number 0 - 9)
- Display also the original image without explanation
To understand what is the name of a label means, we can look at the following figure

To clarify, we don't need to change our UI design, we just need to 1. add the name of the labels, 2. the prediction result and 3. the original image without explanation on it.