Added the new 2 classes model

app.py

@@ -21,7 +21,7 @@ from deepforest import main
 # ------------------------
 # Configuration
 # ------------------------
-YOLO_MODEL_PATH = "yolov8_olive_cls_best.pt"
+YOLO_MODEL_PATH = "olive_cls_2c.pt"
 DEVICE = os.environ.get("DEVICE", "cpu")  # 'cpu' or 'cuda'
 
 def read_imagefile(file_bytes) -> Image.Image:
@@ -204,8 +204,8 @@ async def analyze(image: UploadFile = File(...), conf: float = Form(0.25), iou:
 
         color_map = {
             "healthy": (4, 189, 44),
-            "stressed": (192, 217, 4),
-            "dead": (255, 54, 54),
+            "dry": (192, 217, 4),
+            # "dead": (255, 54, 54),
             "unknown": (128, 128, 128),
         }
 
@@ -248,19 +248,17 @@ async def analyze(image: UploadFile = File(...), conf: float = Form(0.25), iou:
                 states_filled = states.fillna("unknown").astype(str).str.lower()
                 total_trees = int(len(states_filled))
                 healthy_trees = int((states_filled == "healthy").sum())
-                stressed_trees = int((states_filled == "stressed").sum())
-                dead_trees = int((states_filled == "dead").sum())
+                dry_trees = int((states_filled == "dry").sum())
         except Exception:
             total_trees = len(df_pred) if df_pred is not None else 0
-            healthy_trees = stressed_trees = dead_trees = 0
+            healthy_trees = dry_trees = 0
 
         img_b64 = base64.b64encode(buf.getvalue()).decode("ascii")
         return JSONResponse(content={
             "image": img_b64,
             "total_trees_count": total_trees,
             "healthy_trees_count": healthy_trees,
-            "stressed_trees_count": stressed_trees,
-            "dead_trees_count": dead_trees,
+            "dry_trees_count": dry_trees
         })
 
     except Exception as e:

olive_cls_2c.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:89b07b1971e5431a3e7cee676c3c1eab420666a9e1a3b31cf9feb3bf2e034b84
+size 2965505