Create app.py

app.py

@@ -0,0 +1,47 @@
+from fastapi import FastAPI, UploadFile, File
+import tifffile
+import numpy as np
+from cellpose import models
+from huggingface_hub import hf_hub_download
+
+app = FastAPI()
+
+# --------------------------------------------------
+# Download model from HF model repo
+# --------------------------------------------------
+MODEL_REPO = "Shalani08/cellpose_3.1.1_finetuned"
+MODEL_FILE = "ddq_model"
+
+model_path = hf_hub_download(
+    repo_id=MODEL_REPO,
+    filename=MODEL_FILE
+)
+
+# --------------------------------------------------
+# Load Cellpose model (CPU)
+# --------------------------------------------------
+model = models.CellposeModel(
+    pretrained_model=model_path,
+    gpu=False
+)
+
+# --------------------------------------------------
+# API endpoint
+# --------------------------------------------------
+@app.post("/segment")
+async def segment(
+    image: UploadFile = File(...),
+    diameter: float = 0
+):
+    img = tifffile.imread(image.file)
+
+    masks, flows, styles = model.eval(
+        img,
+        diameter=diameter,
+        channels=[0, 0]
+    )
+
+    return {
+        "shape": masks.shape,
+        "masks": masks.tolist()
+    }