Update app.py

app.py

@@ -70,7 +70,7 @@ def main():
 
     if uploaded_file is not None:
         # Read the uploaded image
-        image = cv2.imdecode(np.fromstring(uploaded_file.read(), np.uint8), cv2.IMREAD_GRAYSCALE)
+        image = cv2.imdecode(np.fromstring(uploaded_file.read(), np.uint8), cv2.IMREAD_COLOR)
 
         # Perform inference on the uploaded image
         original_image, mask, segmented_image = perform_inference(image)