power2 commited on
Commit
4e85694
1 Parent(s): 6ea3e3b

user ratinaface inside

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -7,6 +7,7 @@ from networks.layers import AdaIN, AdaptiveAttention
7
  from tensorflow_addons.layers import InstanceNormalization
8
  import numpy as np
9
  import cv2
 
10
  from scipy.ndimage import gaussian_filter
11
 
12
  from tensorflow.keras.models import load_model
@@ -18,7 +19,7 @@ from options.swap_options import SwapOptions
18
  opt = SwapOptions().parse()
19
  token = os.environ['token']
20
 
21
- retina_repo = Repository(local_dir="retina_models", clone_from="felixrosberg/RetinaFace")
22
 
23
  from retinaface.models import *
24
 
 
7
  from tensorflow_addons.layers import InstanceNormalization
8
  import numpy as np
9
  import cv2
10
+ import RetinaFace-Res50.h5
11
  from scipy.ndimage import gaussian_filter
12
 
13
  from tensorflow.keras.models import load_model
 
19
  opt = SwapOptions().parse()
20
  token = os.environ['token']
21
 
22
+ retina_repo = Repository(local_dir="retina_models", RetinaFace-Res50.h5)
23
 
24
  from retinaface.models import *
25