Ahsen Khaliq commited on
Commit
bf1cb4a
1 Parent(s): d9fe6d2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -32,7 +32,6 @@ from argparse import Namespace
32
  from e4e.models.psp import pSp
33
  from util import *
34
 
35
- @ torch.no_grad()
36
  os.makedirs('models', exist_ok=True)
37
  os.system("gdown https://drive.google.com/uc?id=1jtCg8HQ6RlTmLdnbT2PfW1FJ2AYkWqsK")
38
  os.system("cp e4e_ffhq_encode.pt models/e4e_ffhq_encode.pt")
@@ -44,6 +43,7 @@ opts['checkpoint_path'] = model_path
44
  opts= Namespace(**opts)
45
  net = pSp(opts, device).eval().to(device)
46
 
 
47
  def projection(img, name, device='cuda'):
48
 
49
 
 
32
  from e4e.models.psp import pSp
33
  from util import *
34
 
 
35
  os.makedirs('models', exist_ok=True)
36
  os.system("gdown https://drive.google.com/uc?id=1jtCg8HQ6RlTmLdnbT2PfW1FJ2AYkWqsK")
37
  os.system("cp e4e_ffhq_encode.pt models/e4e_ffhq_encode.pt")
 
43
  opts= Namespace(**opts)
44
  net = pSp(opts, device).eval().to(device)
45
 
46
+ @ torch.no_grad()
47
  def projection(img, name, device='cuda'):
48
 
49