Update src/makeanime/cli.py
Browse files- src/makeanime/cli.py +6 -2
src/makeanime/cli.py
CHANGED
|
@@ -37,10 +37,14 @@ def main(
|
|
| 37 |
|
| 38 |
face_image = image if is_gradio else load_image(image)
|
| 39 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 40 |
style_folder = (
|
| 41 |
-
"https://huggingface.co/datasets/ariG23498/images/resolve/main/
|
| 42 |
)
|
| 43 |
-
style_images = [load_image(f"{style_folder}/image00{i}.
|
| 44 |
|
| 45 |
generator = torch.Generator(device="cpu").manual_seed(0)
|
| 46 |
|
|
|
|
| 37 |
|
| 38 |
face_image = image if is_gradio else load_image(image)
|
| 39 |
|
| 40 |
+
# style_folder = (
|
| 41 |
+
# "https://huggingface.co/datasets/ariG23498/images/resolve/main/anime-style"
|
| 42 |
+
# )
|
| 43 |
+
# style_images = [load_image(f"{style_folder}/image00{i}.png") for i in range(10)]
|
| 44 |
style_folder = (
|
| 45 |
+
"https://huggingface.co/datasets/ariG23498/images/resolve/main/sukumar-ray"
|
| 46 |
)
|
| 47 |
+
style_images = [load_image(f"{style_folder}/image00{i}.jpg") for i in range(8)]
|
| 48 |
|
| 49 |
generator = torch.Generator(device="cpu").manual_seed(0)
|
| 50 |
|