Update app.py

app.py

@@ -28,4 +28,4 @@ if upload is not None:
     c2.subheader('Predicted class :')
     c2.write(labels[pred.argmax()])
     c2.subheader('With :')
-    c2.write(f'{int(np.max(pred, axis=1))*100}% assurity')
+    c2.write(f'{int(pred.max())*100}% assurity')