Update app.py

app.py

@@ -11,6 +11,7 @@ from transformers import AutoProcessor, AutoModelForVision2Seq, BitsAndBytesConf
 from transformers.image_utils import load_image
 from peft import PeftModel
 import re
+import spaces
 from diffusers import StableDiffusionXLPipeline, DiffusionPipeline
 
 device = "cuda" if torch.cuda.is_available() else "cpu"