Update app.py

app.py

@@ -15,6 +15,24 @@ with st.spinnet("Unzipping..."):
         zip_ref.extractall(".")
 
 
-st.write(os.listdir(os.getcwd()))
-st.write(os.getcwd())
+# st.write(os.listdir(os.getcwd()))
+# st.write(os.getcwd())
+
+
+model = monai.networks.nets.UNet(
+    in_channels=1,
+    out_channels=3,
+    spatial_dims=3,
+    channels=[16, 32, 64, 128, 256, 512],
+    strides=[2, 2, 2, 2, 2],
+    num_res_units=4,
+    act="PRELU",
+    norm="BATCH",
+    dropout=0.15,
+)
+
+# load this model using anatomy.pt 
+model.load_state_dict(torch.load('anatomy.pt', map_location=torch.device('cpu')))
+
+print(model)