Commit
•
c33cecf
1
Parent(s):
d9a74e2
Update README.md
Browse files
README.md
CHANGED
|
@@ -2,6 +2,8 @@
|
|
| 2 |
tags:
|
| 3 |
- huggan
|
| 4 |
- gan
|
|
|
|
|
|
|
| 5 |
# See a list of available tags here:
|
| 6 |
# https://github.com/huggingface/hub-docs/blob/main/js/src/lib/interfaces/Types.ts#L12
|
| 7 |
# task: unconditional-image-generation or conditional-image-generation or image-to-image
|
|
@@ -33,7 +35,7 @@ transform = Compose(
|
|
| 33 |
Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5)),
|
| 34 |
]
|
| 35 |
)
|
| 36 |
-
model = GeneratorUNet.from_pretrained('huggan/pix2pix-
|
| 37 |
|
| 38 |
def predict_fn(img):
|
| 39 |
inp = transform(img).unsqueeze(0)
|
|
|
|
| 2 |
tags:
|
| 3 |
- huggan
|
| 4 |
- gan
|
| 5 |
+
datasets:
|
| 6 |
+
- huggan/edge2shoes
|
| 7 |
# See a list of available tags here:
|
| 8 |
# https://github.com/huggingface/hub-docs/blob/main/js/src/lib/interfaces/Types.ts#L12
|
| 9 |
# task: unconditional-image-generation or conditional-image-generation or image-to-image
|
|
|
|
| 35 |
Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5)),
|
| 36 |
]
|
| 37 |
)
|
| 38 |
+
model = GeneratorUNet.from_pretrained('huggan/pix2pix-edge2shoes)
|
| 39 |
|
| 40 |
def predict_fn(img):
|
| 41 |
inp = transform(img).unsqueeze(0)
|