Spaces:

AshkanTaghipour
/

HeartWatchAI

Running

Ashkan Taghipour (The University of Western Australia) commited on Feb 9

Commit

2a47411

1 Parent(s): 3ec7be9

Fix model repository names for LVEF models

Files changed (1) hide show

inference.py CHANGED Viewed

@@ -57,8 +57,8 @@ class DeepECGInference:
     # Model repository mappings
     MODEL_REPOS = {
         "diagnosis_77": "heartwise/EfficientNetV2_77_Classes",
-        "lvef_40": "heartwise/EfficientNetV2_LVEF_40",
-        "lvef_50": "heartwise/EfficientNetV2_LVEF_50",
         "afib_5y": "heartwise/EfficientNetV2_AFIB_5y",
     }

     # Model repository mappings
     MODEL_REPOS = {
         "diagnosis_77": "heartwise/EfficientNetV2_77_Classes",
+        "lvef_40": "heartwise/EfficientNetV2_LVEF_equal_under_40",
+        "lvef_50": "heartwise/EfficientNetV2_LVEF_under_50",
         "afib_5y": "heartwise/EfficientNetV2_AFIB_5y",
     }