Update app.py

app.py

@@ -2,6 +2,8 @@ from diffusers import StableDiffusionXLPipeline
 import gradio as gr
 import torch
 
+transformers.utils.move_cache()
+
 def segMindImage(prompt, negative_prompt):
   pipe = StableDiffusionXLPipeline.from_pretrained("segmind/SSD-1B", torch_dtype=torch.float16, use_safetensors=True, variant="fp16")
   pipe.to("cuda")