Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -72,8 +72,8 @@ class EEG_PD(nn.Module):
|
|
| 72 |
# ======================
|
| 73 |
# LOAD MODELS
|
| 74 |
# ======================
|
| 75 |
-
AD_PATH = "
|
| 76 |
-
PD_PATH = "
|
| 77 |
|
| 78 |
print("Loading AD model...")
|
| 79 |
ad_model = EEG_CNN_AD().to(DEVICE)
|
|
|
|
| 72 |
# ======================
|
| 73 |
# LOAD MODELS
|
| 74 |
# ======================
|
| 75 |
+
AD_PATH = "AD_eeg_cnn_ad_ftd_cn.pt"
|
| 76 |
+
PD_PATH = "PD_CNN_FINAL.pth"
|
| 77 |
|
| 78 |
print("Loading AD model...")
|
| 79 |
ad_model = EEG_CNN_AD().to(DEVICE)
|