Update app.py

app.py

@@ -3,33 +3,25 @@ import tifffile
 import numpy as np
 from cellpose import models
 from huggingface_hub import snapshot_download
+import os
 
 app = FastAPI()
 
-# --------------------------------------------------
-# Download full model repo (directory-based model)
-# --------------------------------------------------
 MODEL_REPO = "Shalani08/cellpose_3.1.1_finetuned"
 
 repo_dir = snapshot_download(
     repo_id=MODEL_REPO,
-    repo_type="model"
+    repo_type="model",
+    token=os.environ.get("HF_TOKEN")
 )
 
-# Path to Cellpose model directory
 model_dir = f"{repo_dir}/ddq_model"
 
-# --------------------------------------------------
-# Load Cellpose model (CPU)
-# --------------------------------------------------
 model = models.CellposeModel(
     pretrained_model=model_dir,
     gpu=False
 )
 
-# --------------------------------------------------
-# API endpoint
-# --------------------------------------------------
 @app.post("/segment")
 async def segment(
     image: UploadFile = File(...),
@@ -37,7 +29,7 @@ async def segment(
 ):
     img = tifffile.imread(image.file)
 
-    masks, flows, styles = model.eval(
+    masks, _, _ = model.eval(
         img,
         diameter=diameter,
         channels=[0, 0]