Fix segmentation masks and dependencies

app/segmentation.py

@@ -75,7 +75,7 @@ class SemanticSegmenter:
             img_proc = img.resize(new_size, resample=Image.BILINEAR)
 
         def _split_prompts(text: str) -> list[str]:
-            parts = [p.strip() for p in re.split(r"[;,\\n]", text) if p.strip()]
+            parts = [p.strip() for p in re.split(r"[;,\n]", text) if p.strip()]
             return parts if parts else ([text.strip()] if text.strip() else [])
 
         masks: dict[str, np.ndarray] = {}
@@ -173,6 +173,8 @@ class SegmenterService:
             result["water"] = masks["water"]
         if request.want_road and masks.get("road") is not None:
             result["road"] = masks["road"]
+        if request.want_tree and masks.get("tree") is not None:
+            result["tree"] = masks["tree"]
         return result
 
 

requirements.txt

@@ -1,7 +1,20 @@
+addict
+einops
+fastapi
 gradio>=3.50
+huggingface-hub
+imageio
+matplotlib
 numpy
+omegaconf
 opencv-python
 Pillow
+pillow-heif
+pydantic
 pyyaml
 torch
+torchvision
+tqdm
 transformers>=4.39
+typer
+uvicorn