sxela commited on
Commit
e618b8f
1 Parent(s): 8810a39

fix model path

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