Spaces:
Runtime error
Runtime error
Ahsen Khaliq
commited on
Commit
·
feb9da2
1
Parent(s):
6ac90f7
Update app.py
Browse files
app.py
CHANGED
|
@@ -36,7 +36,7 @@ os.system("mv shape_predictor_68_face_landmarks.dat models/dlibshape_predictor_6
|
|
| 36 |
|
| 37 |
device = 'cpu'
|
| 38 |
|
| 39 |
-
os.system("gdown https://drive.google.com/uc?id=
|
| 40 |
os.system("mv stylegan2-ffhq-config-f.pt models/stylegan2-ffhq-config-f.pt")
|
| 41 |
|
| 42 |
|
|
@@ -60,16 +60,13 @@ transform = transforms.Compose(
|
|
| 60 |
)
|
| 61 |
|
| 62 |
plt.rcParams['figure.dpi'] = 150
|
| 63 |
-
os.system("gdown https://drive.google.com/uc?id=
|
| 64 |
os.system("mv e4e_ffhq_encode.pt models/e4e_ffhq_encode.pt")
|
| 65 |
|
| 66 |
|
| 67 |
-
os.system("gdown https://drive.google.com/uc?id=
|
| 68 |
os.system("mv e4e_ffhq_encode.pt models/jojo.pt")
|
| 69 |
|
| 70 |
-
os.system("gdown https://drive.google.com/uc?id=1ZRwYLRytCEKi__eT2Zxv1IlV6BGVQ_K2")
|
| 71 |
-
os.system("mv e4e_ffhq_encode.pt models/jojo_preserve_color.pt")
|
| 72 |
-
|
| 73 |
def inference(img):
|
| 74 |
filepath = img
|
| 75 |
|
|
|
|
| 36 |
|
| 37 |
device = 'cpu'
|
| 38 |
|
| 39 |
+
os.system("gdown https://drive.google.com/uc?id=1-AG7JPTWc9REBrkll3OyEpZwSOWhlX0j")
|
| 40 |
os.system("mv stylegan2-ffhq-config-f.pt models/stylegan2-ffhq-config-f.pt")
|
| 41 |
|
| 42 |
|
|
|
|
| 60 |
)
|
| 61 |
|
| 62 |
plt.rcParams['figure.dpi'] = 150
|
| 63 |
+
os.system("gdown https://drive.google.com/uc?id=1-7UlCppmiG4DKbhYDNbIZTc6mHy9JMWJ")
|
| 64 |
os.system("mv e4e_ffhq_encode.pt models/e4e_ffhq_encode.pt")
|
| 65 |
|
| 66 |
|
| 67 |
+
os.system("gdown https://drive.google.com/uc?id=1-8E0PFT37v5fZs-61oIrFbNpE28Unp2y")
|
| 68 |
os.system("mv e4e_ffhq_encode.pt models/jojo.pt")
|
| 69 |
|
|
|
|
|
|
|
|
|
|
| 70 |
def inference(img):
|
| 71 |
filepath = img
|
| 72 |
|