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