drop kmeans model into Label Studio, just for testing

README.md

@@ -90,6 +90,13 @@ Application is in `src/label_studio_cyto_ml`
 
 [Setup documentation](src/label_studio_cyto_ml/README.md)
 
+Short version, for testing
+
+```
+cd src
+label-studio-ml start ./label_studio_cyto_ml
+```
+
 ## Pipelines
 
 ### DVC 

scripts/dvc.yaml

@@ -1,6 +1,6 @@
 stages:
-  index:
-    cmd: python image_metadata.py
+  #  index:
+  #  cmd: python image_metadata.py
   embeddings:
     cmd: python image_embeddings.py
     #outs:

src/label_studio_cyto_ml/model.py

@@ -1,3 +1,4 @@
+import pickle
 from typing import Any, Dict, List, Literal, Optional
 
 from label_studio_ml.model import LabelStudioMLBase
@@ -79,7 +80,10 @@ def embeddings_predict(self, embeddings: List[List[float]]) -> List[str]:
         See cyto_ml/visualisation/pages/02_kmeans.py for usage for a collection
         See scripts/cluster.py for the model build and save
         """
-        pass
+        # TODO load this from config, add to Dockerfile
+        fitted = pickle.load(open("../models/kmeans-untagged-images-lana.pkl", "rb"))
+        label = fitted.predict(embeddings)
+        return label
 
     def fit(
         self,