Update README.md
Browse files
README.md
CHANGED
|
@@ -39,7 +39,7 @@ encoded = model.encode(x).sample().mul_(0.18215)
|
|
| 39 |
reconstruction = model.decode(encoded / 0.18215)
|
| 40 |
```
|
| 41 |
|
| 42 |
-
Replace `"VAE-Medical-
|
| 43 |
|
| 44 |
## Citation & Reference
|
| 45 |
|
|
|
|
| 39 |
reconstruction = model.decode(encoded / 0.18215)
|
| 40 |
```
|
| 41 |
|
| 42 |
+
Replace `"VAE-Medical-klf8.pt"` with the desired model filename (`VAE-Medical-klf8.pt` or `VAE-Medical-klf4.pt`).
|
| 43 |
|
| 44 |
## Citation & Reference
|
| 45 |
|