Upload folder using huggingface_hub

README.md

@@ -22,7 +22,10 @@ FastAPI service that serves the MiPLM ESM-2 8M variants as a small REST API.
 | POST   | `/embed`    | `{sequence, model}`               | mean-pooled last-layer embedding         |
 | POST   | `/mutation` | `{sequence, model}`               | wildtype-marginal Δlog-likelihood `[L, 20]` |
 
-`model` is one of `esm2`, `miplm-ce`, `miplm-blosum`. Sequences must be `[A-Za-z]+` and ≤ 1024 residues.
+`model` is one of `miplm-ce` (the cross-entropy 8M variant we trained, surfaced in the
+demo as the "ESM-2" baseline) or `miplm-blosum` (BLOSUM soft-label 8M variant, surfaced
+as "MiPLM"). The MSA variant is not trained yet. Sequences must be `[A-Za-z]+` and
+≤ 1024 residues.
 
 ## Example
 

app.py

@@ -21,7 +21,6 @@ from transformers import EsmForMaskedLM, EsmTokenizer
 
 
 MODEL_REGISTRY: Dict[str, str] = {
-    "esm2":         "facebook/esm2_t6_8M_UR50D",
     "miplm-ce":     "HUBioDataLab/esm2-8m-ce",
     "miplm-blosum": "HUBioDataLab/esm2-8m-softce",
 }