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 |
|