Update app.py

app.py

@@ -2,6 +2,7 @@ import logging
 import random
 import warnings
 import os
+os.environ["GIT_TERMINAL_PROMPT"] = "0"
 import gradio as gr
 import numpy as np
 import spaces
@@ -64,7 +65,7 @@ if not os.path.exists(os.path.join(clip_dir, "t5xxl_fp8_e4m3fn.safetensors")):
 vae_dir = os.path.join(comfy_models_dir, "vae")
 os.makedirs(vae_dir, exist_ok=True)
 if not os.path.exists(os.path.join(vae_dir, "ae.safetensors")):
-    hf_hub_download(repo_id="black-forest-labs/FLUX.1-dev", filename="ae.safetensors", subfolder="vae", local_dir=vae_dir)
+    hf_hub_download(repo_id="black-forest-labs/FLUX.1-dev", filename="ae.safetensors", local_dir=vae_dir)
 
 # Upscale models
 upscale_dir = os.path.join(comfy_models_dir, "upscale_models")