Update app.py

app.py

@@ -67,6 +67,19 @@ from pixel3dmm.run_facer_segmentation import segment
 
 DEVICE = "cuda"
 
+import facer
+from pixel3dmm.lightning.p3dmm_system import system as p3dmm_system
+
+normal_model = p3dmm_system.load_from_checkpoint(f"{env_paths.CKPT_N_PRED}", strict=False)
+normal_model =  normal_model.eval().to(DEVICE)
+
+uv_model = p3dmm_system.load_from_checkpoint(f"{env_paths.CKPT_UV_PRED}", strict=False)
+uv_model =  uv_model.eval().to(DEVICE)
+
+face_detector = facer.face_detector('retinaface/mobilenet', device=DEVICE)
+face_parser   = facer.face_parser   ('farl/celebm/448',     device=DEVICE)
+
+
 # 2. Empty cache for our heavy objects
 _model_cache = {}