Update app.py

app.py

@@ -41,7 +41,7 @@ hf_hub_download(repo_id="black-forest-labs/FLUX.1-dev",
                 local_dir=os.path.join(models_dir, "vae"))
 hf_hub_download(repo_id="black-forest-labs/FLUX.1-dev", 
                 filename="flux1-dev.safetensors", 
-                local_dir=os.path.join(models_dir, "diffusion_models"))
+                local_dir=os.path.join(models_dir, "unet"))
 hf_hub_download(repo_id="google/siglip-so400m-patch14-384", 
                 filename="model.safetensors", 
                 local_dir=os.path.join(models_dir, "clip_vision"))
@@ -91,7 +91,7 @@ with torch.inference_mode():
     # UNET
     unetloader = NODE_CLASS_MAPPINGS["UNETLoader"]()
     UNET_MODEL = unetloader.load_unet(
-        unet_name="flux1-dev.sft",
+        unet_name="flux1-dev.safetensors",
         weight_dtype="fp8_e4m3fn"
     )