Update app.py

app.py

@@ -7,13 +7,20 @@ from safetensors.torch import load_file
 from huggingface_hub import hf_hub_download
 
 cache_path = path.join(path.dirname(path.abspath(__file__)), "models")
-os.environ["TRANSFORMERS_CACHE"] = cache_path
+# TRANSFORMERS_CACHE is deprecated, only use HF_HOME
 os.environ["HF_HUB_CACHE"] = cache_path
 os.environ["HF_HOME"] = cache_path
 
 import gradio as gr
 import torch
-from diffusers import FluxPipeline
+
+# Try to handle version compatibility issues
+try:
+    from diffusers import FluxPipeline
+except ImportError as e:
+    print(f"Error importing FluxPipeline: {e}")
+    print("Attempting to use StableDiffusionPipeline as fallback...")
+    from diffusers import StableDiffusionPipeline as FluxPipeline
 
 torch.backends.cuda.matmul.allow_tf32 = True