fix prediction

src/prediction.py

@@ -33,6 +33,7 @@ def load_model_from_hub():
 
 def predict_image(model, img):
     # Preprocess
+    img = img.convert("RGB")  # VERY IMPORTANT
     img = img.resize((128, 128))
     img_array = np.array(img) / 255.0
     img_array = np.expand_dims(img_array, axis=0)
@@ -66,11 +67,11 @@ def run():
         # To read image file buffer as a PIL Image:
         img = Image.open(uploaded_file)
 
-        image, predicted_class, probs = predict_image(
+        image, predicted_class, confidence = predict_image(
             classifier_model, img
         )
 
-        confidence = np.max(probs) * 100
+        confidence = confidence * 100
         
         # show result
         st.image(img, caption="Uploaded Image", use_container_width=True)