Spaces:

HUBioDataLab
/

miplm-inference

Sleeping

mgorkemuz commited on May 13

Commit

c2a89dc

verified ·

1 Parent(s): c77cd59

Add miplm-msa model to registry

Files changed (1) hide show

app.py CHANGED Viewed

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

 MODEL_REGISTRY: Dict[str, str] = {
     "miplm-ce":     "HUBioDataLab/esm2-8m-ce",
     "miplm-blosum": "HUBioDataLab/esm2-8m-softce",
+    "miplm-msa":    "HUBioDataLab/esm2-8m-msa",
 }
 STANDARD_AAS = "ACDEFGHIKLMNPQRSTVWY"