File size: 882 Bytes
30534a5 c438991 bf105c7 c438991 30534a5 bf105c7 a313617 30534a5 bf105c7 c438991 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 |
---
title: Medical Image Segmentation Gradio App
emoji: 🏥🩺
colorFrom: gray
colorTo: indigo
sdk: gradio
sdk_version: 3.37.0
app_file: app.py
pinned: false
---
# Medical Image Segmentation Gradio App
For the Gradio app we've removed the dependency on pytorch-lightning otherwise used in the project.
The `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).
You 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)
Or via Python:
```python
import wandb
run = wandb.init()
artifact = run.use_artifact('veb-101/UM_medical_segmentation/model-jsr2fn8v:v0', type='model')
artifact_dir = artifact.download()
```
|