import numpy as np import cv2 def preprocess_image(file): # Read bytes from UploadFile image_bytes = file.file.read() # Convert bytes to NumPy array nparr = np.frombuffer(image_bytes, np.uint8) img = cv2.imdecode(nparr, cv2.IMREAD_COLOR) if img is None: raise ValueError("Could not decode image.") img = cv2.resize(img, (256, 256)) # Changed size to 256x256 img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB) img = img / 255.0 img = np.expand_dims(img, axis=0) return img