Tumo505 commited on
Commit
4238a91
·
1 Parent(s): d1ac9b9

Handle space-separated and univariate ECG files (UCR format)

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ import tempfile
15
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
16
 
17
  # Constants
18
- REPO_ID = "Tumo505/SSL-ECG-Classificcation"
19
  CLASS_LABELS = ["NORM", "MI", "STTC", "HYP", "CD"]
20
  CLASS_COLORS = {
21
  "NORM": "#90EE90",
 
15
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
16
 
17
  # Constants
18
+ REPO_ID = "Tumo505/SSL-ECG-Classification-model-card"
19
  CLASS_LABELS = ["NORM", "MI", "STTC", "HYP", "CD"]
20
  CLASS_COLORS = {
21
  "NORM": "#90EE90",