veb-101/UWMGI_Medical_Image_Segmentation

1---2title: Medical Image Segmentation Gradio App3emoji: 🏥🩺4colorFrom: gray5colorTo: indigo6sdk: gradio7sdk_version: 3.37.08app_file: app.py9pinned: false10---11 12# Medical Image Segmentation Gradio App13 14For the Gradio app we've removed the dependency on pytorch-lightning otherwise used in the project.15The `load_lightning_SD_to_Usual_SD.ipynb` notebook contains the steps used to convert pytorch-lightning checkpoint to a regular model checkpoint. This was mainly done to reduce the file size (977 MB --> 244 MB).16 17You can download the original saved checkpoint from over here: [wandb artifact](https://wandb.ai/veb-101/UM_medical_segmentation/artifacts/model/model-jsr2fn8v/v0/files)18 19Or via Python:20 21```python22import wandb23run = wandb.init()24artifact = run.use_artifact('veb-101/UM_medical_segmentation/model-jsr2fn8v:v0', type='model')25artifact_dir = artifact.download()26```27