Update src/generate.py
Browse files- src/generate.py +0 -2
src/generate.py
CHANGED
|
@@ -10,11 +10,9 @@ def generate_face_image(
|
|
| 10 |
) -> Image.Image:
|
| 11 |
"""
|
| 12 |
Generates a synthetic face using the preloaded StyleGAN2 generator.
|
| 13 |
-
|
| 14 |
Args:
|
| 15 |
seed (int): Optional seed for reproducibility.
|
| 16 |
device (str): Device to run generation on ("cuda" or "cpu").
|
| 17 |
-
|
| 18 |
Returns:
|
| 19 |
PIL.Image.Image: Generated face image.
|
| 20 |
"""
|
|
|
|
| 10 |
) -> Image.Image:
|
| 11 |
"""
|
| 12 |
Generates a synthetic face using the preloaded StyleGAN2 generator.
|
|
|
|
| 13 |
Args:
|
| 14 |
seed (int): Optional seed for reproducibility.
|
| 15 |
device (str): Device to run generation on ("cuda" or "cpu").
|
|
|
|
| 16 |
Returns:
|
| 17 |
PIL.Image.Image: Generated face image.
|
| 18 |
"""
|