Update app.py

app.py

@@ -44,7 +44,7 @@ uploaded_file = st.file_uploader("Upload an image of a chest x-ray", type=["jpg"
 if uploaded_file is not None:
     # Convert the file-like object to bytes, then open it with PIL
     image_bytes = uploaded_file.getvalue()
-    image = Image.open(io.BytesIO(image_bytes))
+    image = Image.open(io.BytesIO(image_bytes)).convert('RGB') # make it three channel like training set
     
     # Display the uploaded image
     st.image(image, caption='Uploaded Image.', use_column_width=True)