stemcellai/MorphoAI

1# config.py2import os3 4MODEL_WEIGHTS_PATH = os.getenv(5    "MODEL_WEIGHTS_PATH",6    #"waccbip_t2_model.pth"7    "morphoai_gen2a_model.pth"8)9 10CLASS_MAP = {11    1: "target cell",12    2: "sickle cell",13    3: "elliptocyte",14    4: "echinocyte",15    5: "acanthocyte",16    6: "tear drop",17    7: "stomatocyte",18    8: "helmet cell",19    9: "schistocyte",20}21 22CONDITION_MAP = {23    "target cell": "Thalassemia or cholestatic liver disease",24    "sickle cell": "Sickle cell anemia",25    "elliptocyte": "Hereditary elliptocytosis",26    "echinocyte": "Uremia or liver disease",27    "acanthocyte": "Abetalipoproteinemia",28    "tear drop": "Myelofibrosis",29    "stomatocyte": "Liver disease",30    "helmet cell": "Mechanical trauma",31    "schistocyte": "Microangiopathic hemolytic anemia",32}33 34SCORE_THRESHOLD = float(os.getenv("SCORE_THRESHOLD", 0.5))35