Update app.py

app.py

@@ -3,6 +3,7 @@ from fastapi import FastAPI, File, UploadFile
 from fastapi.responses import StreamingResponse, JSONResponse
 from ndvi_predictor import load_model, normalize_rgb, predict_ndvi, create_visualization
 from PIL import Image
+from io import BytesIO
 import numpy as np
 import io
 import os
@@ -12,7 +13,8 @@ model = load_model("ndvi_best_model.keras")
 
 @app.post("/predict/")
 async def predict_ndvi_api(file: UploadFile = File(...)):
-    img = Image.open(await file.read()).convert("RGB")
+    contents = await file.read()
+    img = Image.open(BytesIO(contents)).convert("RGB")
     rgb_np = np.array(img)
     rgb_norm = normalize_rgb(rgb_np)
     ndvi = predict_ndvi(model, rgb_norm)