GitHub Actions commited on
Commit
0bc21a0
·
1 Parent(s): 2755605

🚀 Deploy embedder from GitHub Actions - 2025-10-27 20:12:19

Browse files
Files changed (1) hide show
  1. embedder.py +2 -2
embedder.py CHANGED
@@ -17,8 +17,8 @@ from open_clip import create_model_and_transforms
17
  MAX_IMAGE_SIZE_MB = 10
18
  MAX_BATCH_SIZE = 10
19
  ALLOWED_EXTENSIONS = {"jpg", "jpeg", "png", "webp"}
20
- MODEL_NAME = "ViT-B-16-256" # MobileCLIP2-S2
21
- PRETRAINED = "datacompxl"
22
 
23
 
24
  # --- Pydantic Models ---
 
17
  MAX_IMAGE_SIZE_MB = 10
18
  MAX_BATCH_SIZE = 10
19
  ALLOWED_EXTENSIONS = {"jpg", "jpeg", "png", "webp"}
20
+ MODEL_NAME = "MobileCLIP2-S2"
21
+ PRETRAINED = "datacomp1b"
22
 
23
 
24
  # --- Pydantic Models ---