Update app.py

app.py

@@ -12,7 +12,7 @@ from scipy.ndimage import gaussian_filter
 from tensorflow.keras.models import load_model
 from options.swap_options import SwapOptions
 
-
+print("hello")
 
 token = os.environ['model_fetch']
 
@@ -20,8 +20,12 @@ opt = SwapOptions().parse()
 
 retina_repo = Repository(local_dir="retina_models", clone_from="felixrosberg/RetinaFace")
 
+print("cloned")
+
 from retina_model.models import *
 
+print("imported")
+
 RetinaFace = load_model("retina_model/RetinaFace-Res50.h5",
                         custom_objects={"FPN": FPN,
                                         "SSH": SSH,
@@ -29,6 +33,8 @@ RetinaFace = load_model("retina_model/RetinaFace-Res50.h5",
                                         "LandmarkHead": LandmarkHead,
                                         "ClassHead": ClassHead})
 
+print("loading model")
+
 arc_repo = Repository(local_dir="arcface_model", clone_from="felixrosberg/ArcFace")
 ArcFace = load_model("arcface_model/ArcFace-Res50.h5")
 ArcFaceE = load_model("arcface_model/ArcFacePerceptual-Res50.h5")