sxela commited on
Commit
91b67a1
1 Parent(s): 04a9d78

bump to 0.4

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -3,7 +3,7 @@ Thanks to nateraw for making this scape happen!
3
  This code has been mostly taken from https://huggingface.co/spaces/nateraw/animegan-v2-for-videos/tree/main
4
  """
5
  import os
6
- os.system("wget https://github.com/Sxela/ArcaneGAN/releases/download/v0.3/ArcaneGANv0.3.jit")
7
 
8
  import sys
9
  from subprocess import call
@@ -34,7 +34,8 @@ from torchvision.transforms.functional import center_crop, to_tensor
34
 
35
 
36
  print("🧠 Loading Model...")
37
- model = torch.jit.load('./ArcaneGANv0.3.jit').cuda().eval().half()
 
38
 
39
  # This function is taken from pytorchvideo!
40
  def uniform_temporal_subsample(x: torch.Tensor, num_samples: int, temporal_dim: int = -3) -> torch.Tensor:
 
3
  This code has been mostly taken from https://huggingface.co/spaces/nateraw/animegan-v2-for-videos/tree/main
4
  """
5
  import os
6
+ os.system("wget https://github.com/Sxela/ArcaneGAN/releases/download/v0.3/ArcaneGANv0.4.jit")
7
 
8
  import sys
9
  from subprocess import call
 
34
 
35
 
36
  print("🧠 Loading Model...")
37
+ #model = torch.jit.load('./ArcaneGANv0.3.jit').cuda().eval().half()
38
+ model = torch.jit.load('./ArcaneGANv0.4.jit').cuda().eval().half()
39
 
40
  # This function is taken from pytorchvideo!
41
  def uniform_temporal_subsample(x: torch.Tensor, num_samples: int, temporal_dim: int = -3) -> torch.Tensor: