Update app.py

app.py

@@ -26,22 +26,27 @@ async def root():
 @app.post("/predict")
 async def predict(file: UploadFile = File(...)):
     try:
+        # Load and preprocess image
         image = Image.open(file.file).convert("RGB").resize((224, 224))
         img_array = np.expand_dims(np.array(image) / 255.0, axis=0)
 
+        # Predict
         prediction = model.predict(img_array)
         predicted_class = int(np.argmax(prediction))
         confidence = float(np.max(prediction))
 
-        # 🔹 Map class index to cancer type name
+        # 🔹 Map class index to cell type
         classes = [
+            "Basophil",
             "Eosinophil",
+            "Erythroblast",
+            "Immature Granulocyte (IG)",
             "Lymphocyte",
             "Monocyte",
             "Neutrophil",
-            "Cancerous Cell",
-            "Unknown"
+            "Platelet"
         ]
+
         label = classes[predicted_class]
 
         return JSONResponse(