Update app.py

app.py

@@ -13,11 +13,11 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
 MAX_SEED = np.iinfo(np.int32).max
 
 # Model repository ID for 9B distilled
-REPO_ID_DISTILLED = "black-forest-labs/FLUX.2-klein-9B"
+REPO_ID_DISTILLED = "black-forest-labs/FLUX.2-klein-4B"
 
 # LoRA repository and file
 LORA_REPO_ID = "mr2along/BFS"
-LORA_FILENAME = "bfs_head_v1_flux-klein_9b_step3750_rank64.safetensors"
+LORA_FILENAME = "bfs_head_v1_flux-klein_4b.safetensors"
 
 # Fixed prompt for face swapping
 #FACE_SWAP_PROMPT = "head_swap: start with Picture 1 as the base image, keeping its lighting, environment, and background. remove the head from Picture 1 completely and replace it with the head from Picture 2, strictly preserving the hair, eye color, nose structure of Picture 2. copy the direction of the eye, head rotation, micro expressions from Picture 1, high quality, sharp details, 4k."