Update app.py

app.py

@@ -40,12 +40,11 @@ st.markdown(custom_style, unsafe_allow_html=True)
 
 
 # Function to perform inference
-def perform_inference(img):
-    image = cv2.imread(img, cv2.IMREAD_COLOR)
+def perform_inference(image):
     original_shape = image.shape[:2]
     original_image = image.copy()
     image = cv2.resize(image, (W, H))
-    image = image/255.0
+    image = image / 255.0
     image = np.expand_dims(image, axis=0)
 
     mask = model.predict(image, verbose=0)[0]
@@ -70,8 +69,11 @@ def main():
     uploaded_file = st.file_uploader("Upload a brain tumor image...", type=["jpg", "png", "jpeg"])
 
     if uploaded_file is not None:
+        # Read the uploaded image
+        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(uploaded_file)
+        original_image, mask, segmented_image = perform_inference(image)
 
         # Display original image
         st.subheader("Original Image")