Update app.py

app.py

@@ -10,9 +10,25 @@ from PIL import Image
 
 import torch
 
-
 print(f'torch version:{torch.__version__}')
 
+import subprocess
+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()
+
+def sh(cmd): subprocess.check_call(cmd, shell=True)
+
+sh("pip install -U xformers --index-url https://download.pytorch.org/whl/cu126")
+
+# tell Python to re-scan site-packages now that the egg-link exists
+import importlib, site; site.addsitedir(site.getsitepackages()[0]); importlib.invalidate_caches()
+
 import torch.utils.checkpoint
 from pytorch_lightning import seed_everything
 from diffusers import AutoencoderKL, DDPMScheduler