Update app.py

app.py

@@ -9,6 +9,14 @@ import time
 import gradio as gr
 import sys
 from PIL import Image
+import importlib, site, sys
+
+# Re-discover all .pth/.egg-link files
+for sitedir in site.getsitepackages():
+    site.addsitedir(sitedir)
+
+# Clear caches so importlib will pick up new modules
+importlib.invalidate_caches()
 
 # Set environment variables
 os.environ["PIXEL3DMM_CODE_BASE"] = f"{os.getcwd()}"
@@ -19,8 +27,12 @@ def sh(cmd): subprocess.check_call(cmd, shell=True)
 
 sh("pip install -e .")
 
+# tell Python to re-scan site-packages now that the egg-link exists
+import importlib, site; site.addsitedir(site.getsitepackages()[0]); importlib.invalidate_caches()
+
 from pixel3dmm import env_paths
 
+
 sh("cd src/pixel3dmm/preprocessing/facer && pip install -e . && cd ../../../..")
 sh("cd src/pixel3dmm/preprocessing/PIPNet/FaceBoxesV2/utils && sh make.sh && cd ../../../../../..")