xf3227 commited on
Commit
447e709
·
verified ·
1 Parent(s): 780caa8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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 = './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'
 
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'