felixrosberg commited on
Commit
ea104c0
β€’
1 Parent(s): e921287

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -10,7 +10,6 @@ import cv2
10
  from scipy.ndimage import gaussian_filter
11
 
12
  from tensorflow.keras.models import load_model
13
- from retinaface.models import *
14
  from options.swap_options import SwapOptions
15
 
16
  # Invalidated!
@@ -21,6 +20,8 @@ opt = SwapOptions().parse()
21
 
22
  retina_repo = Repository(local_dir="retina_model", clone_from="felixrosberg/retinaface_resnet50",
23
  private=True, use_auth_token=token, git_user="felixrosberg")
 
 
24
  RetinaFace = load_model("retina_model/retinaface_res50.h5",
25
  custom_objects={"FPN": FPN,
26
  "SSH": SSH,
 
10
  from scipy.ndimage import gaussian_filter
11
 
12
  from tensorflow.keras.models import load_model
 
13
  from options.swap_options import SwapOptions
14
 
15
  # Invalidated!
 
20
 
21
  retina_repo = Repository(local_dir="retina_model", clone_from="felixrosberg/retinaface_resnet50",
22
  private=True, use_auth_token=token, git_user="felixrosberg")
23
+
24
+ from retina_model.models import *
25
  RetinaFace = load_model("retina_model/retinaface_res50.h5",
26
  custom_objects={"FPN": FPN,
27
  "SSH": SSH,