yourusername commited on
Commit
35d1acb
1 Parent(s): a29ba80

:memo: add print statement

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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)