Update app.py

app.py

@@ -33,15 +33,20 @@ from e4e.models.psp import pSp
 from util import *
 
 @ torch.no_grad()
+os.makedirs('models', exist_ok=True)
+os.system("gdown https://drive.google.com/uc?id=1jtCg8HQ6RlTmLdnbT2PfW1FJ2AYkWqsK")
+os.system("cp e4e_ffhq_encode.pt models/e4e_ffhq_encode.pt")
+
+model_path = 'models/e4e_ffhq_encode.pt'
+ckpt = torch.load(model_path, map_location='cpu')
+opts = ckpt['opts']
+opts['checkpoint_path'] = model_path
+opts= Namespace(**opts)
+net = pSp(opts, device).eval().to(device)
 
 def projection(img, name, device='cuda'):
  
-    model_path = 'models/e4e_ffhq_encode.pt'
-    ckpt = torch.load(model_path, map_location='cpu')
-    opts = ckpt['opts']
-    opts['checkpoint_path'] = model_path
-    opts= Namespace(**opts)
-    net = pSp(opts, device).eval().to(device)
+    
     transform = transforms.Compose(
         [
             transforms.Resize(256),
@@ -57,9 +62,7 @@ def projection(img, name, device='cuda'):
     torch.save(result_file, name)
     return w_plus[0]
 
-os.makedirs('models', exist_ok=True)
-os.system("gdown https://drive.google.com/uc?id=1jtCg8HQ6RlTmLdnbT2PfW1FJ2AYkWqsK")
-os.system("cp e4e_ffhq_encode.pt models/e4e_ffhq_encode.pt")
+
 
 os.system("wget http://dlib.net/files/shape_predictor_68_face_landmarks.dat.bz2")
 os.system("bzip2 -dk shape_predictor_68_face_landmarks.dat.bz2")