Update app.py

app.py

@@ -22,13 +22,13 @@ st.markdown("To use the demo:\n* Provide input features in the form below. Featu
 @st.cache_resource
 def load_model():
     import adrd
-    try:   
-        ckpt_path = '/home/user/app/ckpt_swinunetr_stripped_MNI.pt'
-        model = adrd.model.ADRDModel.from_ckpt(ckpt_path, device='cpu')
-    except:
-        # ckpt_path = '../adrd_tool_copied_from_sahana/dev/ckpt/ckpt_swinunetr_stripped_MNI.pt'
-        ckpt_path = '/data_1/skowshik/ckpts_backbone_swinunet/ckpt_swinunetr_stripped_MNI.pt'
-        model = adrd.model.ADRDModel.from_ckpt(ckpt_path, device='cpu')
+    # try:   
+    # except:
+    #     # ckpt_path = '../adrd_tool_copied_from_sahana/dev/ckpt/ckpt_swinunetr_stripped_MNI.pt'
+    #     ckpt_path = '/data_1/skowshik/ckpts_backbone_swinunet/ckpt_swinunetr_stripped_MNI.pt'
+    #     model = adrd.model.ADRDModel.from_ckpt(ckpt_path, device='cpu')
+    ckpt_path = '/home/user/app/ckpt_swinunetr_stripped_MNI.pt'
+    model = adrd.model.ADRDModel.from_ckpt(ckpt_path, device='cpu')
     return model
 
 model = load_model()