Update app.py
Browse files
app.py
CHANGED
@@ -23,7 +23,7 @@ st.markdown("To use the demo:\n* Provide input features in the form below. Featu
|
|
23 |
def load_model():
|
24 |
import adrd
|
25 |
try:
|
26 |
-
ckpt_path = '
|
27 |
model = adrd.model.ADRDModel.from_ckpt(ckpt_path, device='cpu')
|
28 |
except:
|
29 |
# ckpt_path = '../adrd_tool_copied_from_sahana/dev/ckpt/ckpt_swinunetr_stripped_MNI.pt'
|
|
|
23 |
def load_model():
|
24 |
import adrd
|
25 |
try:
|
26 |
+
ckpt_path = '/home/user/app/ckpt_swinunetr_stripped_MNI.pt'
|
27 |
model = adrd.model.ADRDModel.from_ckpt(ckpt_path, device='cpu')
|
28 |
except:
|
29 |
# ckpt_path = '../adrd_tool_copied_from_sahana/dev/ckpt/ckpt_swinunetr_stripped_MNI.pt'
|