Update app.py (#11)

- Update app.py (ef9744a38c2043acf8a0e90c24ada00c8b1fea72)


Co-authored-by: Aditi Baheti <Aditibaheti@users.noreply.huggingface.co>

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)