Update
Browse files
app.py
CHANGED
|
@@ -63,6 +63,7 @@ mediapy.set_ffmpeg(ffmpeg_path)
|
|
| 63 |
|
| 64 |
config = OmegaConf.load("configs/pseudoseg000.yaml")
|
| 65 |
config.model.load_pretrained = "seg.ckpt"
|
|
|
|
| 66 |
seg_model = builder.build_model(config).eval()
|
| 67 |
examples = glob.glob("examples/*.nii.gz")
|
| 68 |
|
|
@@ -76,6 +77,6 @@ with gr.Blocks(theme="dark-peach") as demo:
|
|
| 76 |
|
| 77 |
|
| 78 |
if __name__ == "__main__":
|
| 79 |
-
demo.launch(debug=True
|
| 80 |
|
| 81 |
|
|
|
|
| 63 |
|
| 64 |
config = OmegaConf.load("configs/pseudoseg000.yaml")
|
| 65 |
config.model.load_pretrained = "seg.ckpt"
|
| 66 |
+
config.model.params.encoder_params.pretrained = False
|
| 67 |
seg_model = builder.build_model(config).eval()
|
| 68 |
examples = glob.glob("examples/*.nii.gz")
|
| 69 |
|
|
|
|
| 77 |
|
| 78 |
|
| 79 |
if __name__ == "__main__":
|
| 80 |
+
demo.launch(debug=True)
|
| 81 |
|
| 82 |
|