Update app.py

app.py

@@ -160,7 +160,7 @@ def load_models() -> str | None:
         # Load TenStrip fine-tuned checkpoint into the transformer
         from huggingface_hub import hf_hub_download
         from safetensors.torch import load_model
-        finetune_path = hf_hub_download(repo_id="TenStrip/10Eros-Max", filename="10Eros_Max_h3_fl2va_bf16_test1.safetensors")
+        finetune_path = hf_hub_download(repo_id="TenStrip/10Eros-Max", filename="10Eros_Max_h3_fl2va_bf16_test2_pruned.safetensors")
         load_model(pipe.transformer, finetune_path, strict=False, device="cpu")
         pipe.transformer.to(torch.bfloat16)
         pipe.transformer.set_attention_backend(ATTENTION)