akhaliq HF staff commited on
Commit
c41263f
1 Parent(s): 5b2ed41

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -40,7 +40,7 @@ device= 'cpu'
40
  model_path_e = hf_hub_download(repo_id="akhaliq/JoJoGAN_e4e_ffhq_encode", filename="e4e_ffhq_encode.pt")
41
  ckpt = torch.load(model_path_e, map_location='cpu')
42
  opts = ckpt['opts']
43
- opts['checkpoint_path'] = model_path
44
  opts= Namespace(**opts)
45
  net = pSp(opts, device).eval().to(device)
46
 
40
  model_path_e = hf_hub_download(repo_id="akhaliq/JoJoGAN_e4e_ffhq_encode", filename="e4e_ffhq_encode.pt")
41
  ckpt = torch.load(model_path_e, map_location='cpu')
42
  opts = ckpt['opts']
43
+ opts['checkpoint_path'] = model_path_e
44
  opts= Namespace(**opts)
45
  net = pSp(opts, device).eval().to(device)
46