🚀 Deploy embedder from GitHub Actions - 2025-10-27 20:24:18

Dockerfile

@@ -33,7 +33,9 @@ RUN pip install --no-cache-dir --upgrade pip && \
     pip install --no-cache-dir --extra-index-url https://download.pytorch.org/whl/cpu \
       torch torchvision && \
     pip install --no-cache-dir \
-      open_clip_torch
+      open_clip_torch && \
+    pip install --no-cache-dir \
+      git+https://github.com/apple/ml-mobileclip.git
 
 
 # Copy application code

embedder.py

@@ -11,6 +11,7 @@ from fastapi.responses import JSONResponse
 from PIL import Image
 from pydantic import BaseModel, Field
 from open_clip import create_model_and_transforms
+from mobileclip.modules.common.mobileone import reparameterize_model
 
 
 # --- Configuration ---
@@ -18,7 +19,7 @@ MAX_IMAGE_SIZE_MB = 10
 MAX_BATCH_SIZE = 10
 ALLOWED_EXTENSIONS = {"jpg", "jpeg", "png", "webp"}
 MODEL_NAME = "MobileCLIP2-S2"
-PRETRAINED = "datacomp1b"
+PRETRAINED = "dfndr2b"
 
 
 # --- Pydantic Models ---
@@ -89,6 +90,10 @@ def load_model():
     model = model.to(device)
     model.eval()
     
+    # Reparameterize model for inference (required for MobileCLIP)
+    print("Reparameterizing model for inference...")
+    model = reparameterize_model(model)
+    
     print(f"✓ Model loaded: {MODEL_NAME}")
     print(f"✓ Pretrained: {PRETRAINED}")
     print(f"✓ Device: {device}")

requirements.txt

@@ -5,4 +5,5 @@ pillow
 numpy
 pydantic
 torch
-open_clip_torch
+open_clip_torch
+ml-mobileclip @ git+https://github.com/apple/ml-mobileclip.git