Update README.md
Browse files
README.md
CHANGED
|
@@ -27,11 +27,9 @@ The models proposed in the paper _"TryOffDiff: Virtual-Try-Off via High-Fidelity
|
|
| 27 |
## Usage
|
| 28 |
|
| 29 |
```python
|
| 30 |
-
from torchvision.io import read_image
|
| 31 |
-
from torchvision.models.detection import MaskRCNN_ResNet50_FPN_Weights
|
| 32 |
from huggingface_hub import hf_hub_download
|
| 33 |
|
| 34 |
-
|
| 35 |
repo_id="rizavelioglu/tryoffdiff",
|
| 36 |
filename="tryoffdiff.pth", # or one of ["ldm-1", "ldm-2", "ldm-3", ...]
|
| 37 |
)
|
|
|
|
| 27 |
## Usage
|
| 28 |
|
| 29 |
```python
|
|
|
|
|
|
|
| 30 |
from huggingface_hub import hf_hub_download
|
| 31 |
|
| 32 |
+
path_model = hf_hub_download(
|
| 33 |
repo_id="rizavelioglu/tryoffdiff",
|
| 34 |
filename="tryoffdiff.pth", # or one of ["ldm-1", "ldm-2", "ldm-3", ...]
|
| 35 |
)
|