Update app.py

app.py

@@ -41,8 +41,8 @@ os.environ["XVERSE_PREPROCESSED_DATA"] = f"{os.getcwd()}/proprocess_data"
 
 # FLUX.1-schnell
 snapshot_download(
-    repo_id="black-forest-labs/FLUX.1-dev",
-    local_dir="./checkpoints/FLUX.1-dev",
+    repo_id="black-forest-labs/FLUX.1-schnell",
+    local_dir="./checkpoints/FLUX.1-schnell",
     local_dir_use_symlinks=False
 )
 
@@ -94,7 +94,7 @@ os.environ["FLORENCE2_MODEL_PATH"]    = "./checkpoints/Florence-2-large"
 os.environ["SAM2_MODEL_PATH"]         = "./checkpoints/sam2.1_hiera_large.pt"
 os.environ["FACE_ID_MODEL_PATH"]      = "./checkpoints/model_ir_se50.pth"
 os.environ["CLIP_MODEL_PATH"]         = "./checkpoints/clip-vit-large-patch14"
-os.environ["FLUX_MODEL_PATH"]         = "./checkpoints/FLUX.1-dev"
+os.environ["FLUX_MODEL_PATH"]         = "./checkpoints/FLUX.1-schnell"
 os.environ["DPG_VQA_MODEL_PATH"]      = "./checkpoints/mplug_visual-question-answering_coco_large_en"
 os.environ["DINO_MODEL_PATH"]         = "./checkpoints/dino-vits16"