Update app.py

app.py

@@ -5,9 +5,16 @@ from diffusers import DiffusionPipeline
 import torch
 from huggingface_hub import login
 import os
+import subprocess
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
+# Install xformers if not already installed
+try:
+    import xformers
+except ImportError:
+    subprocess.check_call([sys.executable, "-m", "pip", "install", "xformers"])
+
 # Set your Hugging Face token
 HUGGINGFACE_TOKEN = os.getenv("HUGGINGFACE_TOKEN")
 login(token=HUGGINGFACE_TOKEN)