Ahsen Khaliq commited on
Commit
feb9da2
1 Parent(s): 6ac90f7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -6
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=1X6AvjLltUAYZrarLdrdGav5gsRkqJYOE")
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=1o6ijA3PkcewZvwJJ73dJ0fxhndn0nnh7")
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=13cR2xjIBj8Ga5jMO7gtxzIJj2PDsBYK4")
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