Spaces:

ruanhelansantos
/

EndtoEnd_AI_Solution_Implementation

Sleeping

ruanhelansantos commited on Mar 1

Commit

fe3003e

verified ·

1 Parent(s): 8708588

Update model.py

Files changed (1) hide show

model.py CHANGED Viewed

@@ -16,8 +16,8 @@ PROCESSOR_NAME = "google/vit-base-patch16-224"
 #   - Known STEMI ECG     → top prediction: index 0  → MI   ✓
 # Ordering: MI, NORM, STTC, CD, AFIB (consistent with PTB-XL fine-tune conventions)
 LABEL_MAP = {
-    0: "MI",
-    1: "NORM",
     2: "STTC",
     3: "CD",
     4: "AFIB",

 #   - Known STEMI ECG     → top prediction: index 0  → MI   ✓
 # Ordering: MI, NORM, STTC, CD, AFIB (consistent with PTB-XL fine-tune conventions)
 LABEL_MAP = {
+    0: "NORM",
+    1: "MI",
     2: "STTC",
     3: "CD",
     4: "AFIB",