No, you don’t need to retrain the model if you just want to get the probabilities with a softmax layer.
I think the easiest way could be to pass your current model and the nn.Softmax layer to a new custom model or an nn.Sequential container and use this new combined model in your app.
Alternatively, you could compute the softmax on the mobile, as the formula is quite straightforward.