update

app.py

@@ -25,11 +25,12 @@ def load_model():
         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'
-        # model = adrd.model.ADRDModel.from_ckpt(ckpt_path, device='cpu')
-        return None
+        ckpt_path = '/data_1/skowshik/ckpts_backbone_swinunet/ckpt_swinunetr_stripped_MNI.pt'
+        model = adrd.model.ADRDModel.from_ckpt(ckpt_path, device='cpu')
     return model
 
 model = load_model()
+print(dir(model))
 
 def predict_proba(data_dict):
     pred_dict = model.predict_proba([data_dict])[1][0]
@@ -103,7 +104,6 @@ def create_input(i):
             st.number_input(description, key=name, min_value=min_value, value=default_value, placeholder=values['range'])
     else:
         values = {int(k): v for k, v in values.items()}
-        reverse_mapping = {v: k for k, v in values.items()}
         if default_value in values:
             default_index = list(values.keys()).index(default_value)
         else:
@@ -166,5 +166,12 @@ if predict_button:
     data_dict = convert_dictionary(data_dict, nacc_mapping)
     pred_dict = predict_proba(data_dict)
 
+    # change key name
+    # key_mappings = {
+    #     'NC': 'Normal cognition',
+    #     'MCI':
+
+    # }
+
     st.write("Predicted probabilities:")
     st.code(json.dumps(pred_dict, indent=2))