Update app.py

app.py

@@ -58,8 +58,15 @@ def preprocess_image(file_path):
 def binary_decision(confidence):
     return 1 if confidence >= 0.5 else 0
 
-def predict_braintumor(img_path):
-    img_gray = cv2.imread(img_path, cv2.IMREAD_GRAYSCALE)
+def predict_braintumor(file_contents):
+    # Save the uploaded file
+    filename = "temp_image.png"
+    file_path = os.path.join(UPLOAD_FOLDER, filename)
+
+    with open(file_path, "wb") as f:
+        f.write(file_contents)
+
+    img_gray = cv2.imread(file_path, cv2.IMREAD_GRAYSCALE)
 
     # Crop and preprocess the grayscale image
     img_processed = crop_imgs([img_gray])
@@ -85,15 +92,8 @@ def main():
         # Read the contents of the uploaded file
         file_contents = uploaded_file.read()
 
-        # Save the uploaded file
-        filename = secure_filename(uploaded_file.name)
-        file_path = os.path.join(UPLOAD_FOLDER, filename)
-
-        with open(file_path, "wb") as f:
-            f.write(file_contents)
-
         # Make prediction
-        result = predict_braintumor(file_path)
+        result = predict_braintumor(file_contents)
 
         # Display prediction
         st.subheader("Prediction:")