Update app.py

app.py

@@ -6,7 +6,7 @@ import torch
 from PIL import Image
 import spaces
 from diffusers import DiffusionPipeline, AutoPipelineForText2Image
-from diffusers import StableDiffusion3Pipeline # pip install diffusers>=0.31.0
+from diffusers import StableDiffusion3Pipeline, FlowMatchEulerDiscreteScheduler, SD3Transformer2DModel # pip install diffusers>=0.31.0
 import copy
 import random
 import time
@@ -14,6 +14,11 @@ from huggingface_hub import login
 hf_token = os.environ.get("HF_TOKEN")
 login(token=hf_token)
 
+cache_path = path.join(path.dirname(path.abspath(__file__)), "models")
+os.environ["TRANSFORMERS_CACHE"] = cache_path
+os.environ["HF_HUB_CACHE"] = cache_path
+os.environ["HF_HOME"] = cache_path
+
 torch.set_float32_matmul_precision("high")
 
 #torch._inductor.config.conv_1x1_as_mm = True