Spaces:
Runtime error
Runtime error
Ahsen Khaliq
commited on
Commit
•
5866cd5
1
Parent(s):
bf1cb4a
Update app.py
Browse files
app.py
CHANGED
@@ -36,6 +36,7 @@ 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")
|
38 |
|
|
|
39 |
model_path = 'models/e4e_ffhq_encode.pt'
|
40 |
ckpt = torch.load(model_path, map_location='cpu')
|
41 |
opts = ckpt['opts']
|
|
|
36 |
os.system("gdown https://drive.google.com/uc?id=1jtCg8HQ6RlTmLdnbT2PfW1FJ2AYkWqsK")
|
37 |
os.system("cp e4e_ffhq_encode.pt models/e4e_ffhq_encode.pt")
|
38 |
|
39 |
+
device= 'cpu'
|
40 |
model_path = 'models/e4e_ffhq_encode.pt'
|
41 |
ckpt = torch.load(model_path, map_location='cpu')
|
42 |
opts = ckpt['opts']
|