Update app.py

app.py

@@ -1,3 +1,4 @@
+import streamlit as st
 import cv2
 import numpy as np
 import tensorflow as tf
@@ -122,11 +123,6 @@ if uploaded_file is not None:
     predictions = model.predict(processed_image)
     probabilities = predictions[0]
 
-    # Display probabilities for each class
-    for i, label in enumerate(class_labels):
-        if probabilities[i] > 0:
-            st.write(f"{label}: {probabilities[i]:.2f}")
-
     # Show predicted class
     predicted_class = class_labels[np.argmax(probabilities)]
     st.write(f"Classe prédite: {predicted_class}")
@@ -138,4 +134,4 @@ if uploaded_file is not None:
         if treatment_info['medicines']:
             st.write("Médicaments recommandés:", ', '.join(treatment_info['medicines']))
         else:
-            st.write("Aucun médicament requis.")
+            st.write("Aucun médicament requis.")