convert to RGB

app.py

@@ -59,6 +59,7 @@ def on_click():
     else:
         image = Image.open(file)
         st.image(image, use_column_width=True)
+        image = image.convert('RGB')
         predictions = model_prediction(np.array(image))
         st.write(prediction)
         print(