Spaces:
Sleeping
Sleeping
yourusername
commited on
Commit
·
35d1acb
1
Parent(s):
a29ba80
:memo: add print statement
Browse files
app.py
CHANGED
@@ -10,6 +10,7 @@ from encoded_video import EncodedVideo, write_video
|
|
10 |
from PIL import Image
|
11 |
from torchvision.transforms.functional import center_crop, to_tensor
|
12 |
|
|
|
13 |
model = torch.hub.load(
|
14 |
"AK391/animegan2-pytorch:main",
|
15 |
"generator",
|
@@ -18,7 +19,6 @@ model = torch.hub.load(
|
|
18 |
progress=True,
|
19 |
)
|
20 |
|
21 |
-
|
22 |
def face2paint(model: torch.nn.Module, img: Image.Image, size: int = 512, device: str = 'cuda'):
|
23 |
w, h = img.size
|
24 |
s = min(w, h)
|
|
|
10 |
from PIL import Image
|
11 |
from torchvision.transforms.functional import center_crop, to_tensor
|
12 |
|
13 |
+
print("🧠 Loading Model...")
|
14 |
model = torch.hub.load(
|
15 |
"AK391/animegan2-pytorch:main",
|
16 |
"generator",
|
|
|
19 |
progress=True,
|
20 |
)
|
21 |
|
|
|
22 |
def face2paint(model: torch.nn.Module, img: Image.Image, size: int = 512, device: str = 'cuda'):
|
23 |
w, h = img.size
|
24 |
s = min(w, h)
|