Update app.py

app.py

@@ -46,13 +46,14 @@ elif selection == "Application":
 				if btn_prediction:
 					predictions = model.predict(image_array)
 					classes = np.argmax(predictions[0])
+                    prediction_proba = model.predict_proba(image_array)
 
 					for cle, valeur in covid_classes.items():
 						if valeur == classes:
 							st.write("The diagostic is :",cle)
 
-		#Donner un titre
-
+                		st.subheader('Prediction Probability')
+                        st.write(prediction_proba)
 	if __name__ == '__main__':
 		main()