Update app.py

app.py

@@ -9,7 +9,7 @@ nn = []
 
 # Fonction pour traiter et redimensionner l'image importée
 def process_uploaded_image(uploaded_file):
-    image = Image.open(uploaded_file).resize((300, 300))
+    image = Image.open(uploaded_file).convert("RGB").resize((300, 300))  # Conversion en RGB
     pixel_data = np.array(image)
     return pixel_data
 
@@ -60,13 +60,13 @@ if uploaded_file:
         if st.button("Oui"):
             # Enregistrer les pixels de l'image générée dans la liste "oe"
             oe.extend(generated_image.getdata())
-            oe = list(set(oe))  # Enlever les doublons
+            oe = list(set(map(tuple, oe)))  # Enlever les doublons
             st.success("Image validée!")
         
         elif st.button("Non"):
             # Enregistrer les pixels de l'image générée dans la liste "nn"
             nn.extend(generated_image.getdata())
-            nn = list(set(nn))  # Enlever les doublons
+            nn = list(set(map(tuple, nn)))  # Enlever les doublons
             st.error("Image rejetée!")
 
     # Affichage des listes de pixels validés et non validés