Update app.py

app.py

@@ -15,7 +15,7 @@ LORA_VERSION = "DatasetTest4.safetensors"
 # Download required models
 t5_path = hf_hub_download(repo_id="silveroxides/flan-t5-xxl-encoder-only", filename="flan-t5-xxl_float8_e4m3fn_scaled_stochastic.safetensors", local_dir="models/text_encoders/")
 vae_path = hf_hub_download(repo_id="lodestones/Chroma", filename="ae.safetensors", local_dir="models/vae")
-unet_path = hf_hub_download(repo_id="silveroxides/Chroma2-Kaleidoscope", filename=CHROMA_VERSION, local_dir="models/unet")
+unet_path = hf_hub_download(repo_id="lodestones/Chroma2-Kaleidoscope", filename=CHROMA_VERSION, local_dir="models/unet")
 lora_path = hf_hub_download(repo_id="Alme995/DatasetTest_Loras_Chroma", filename=LORA_VERSION, local_dir="models/loras")
 
 # Example prompts with their parameters