Spaces:
Runtime error
Runtime error
Ahsen Khaliq
commited on
Commit
•
bf1cb4a
1
Parent(s):
d9fe6d2
Update app.py
Browse files
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 |
|