Update app.py

app.py

@@ -31,6 +31,7 @@ from src.utils.camera_util import (
 )
 from src.utils.mesh_util import save_obj, save_glb
 from src.utils.infer_util import remove_background, resize_foreground, images_to_video
+from huggingface_hub import hb
 
 # Set up cache path
 cache_path = path.join(path.dirname(path.abspath(__file__)), "models")
@@ -75,7 +76,8 @@ else:
 device = torch.device('cuda')
 
 base_model = "black-forest-labs/FLUX.1-dev"
-pipe = FluxPipeline.from_single_file("https://huggingface.co/marduk191/Flux.1_collection/resolve/main/flux.1_dev_fp8_fp16t5-marduk191.safetensors", torch_dtype=torch.bfloat16, token=huggingface_token).to(device)
+file_flux = hf_hub_download("marduk191/Flux.1_collection", "flux.1_dev_fp8_fp16t5-marduk191.safetensors")
+pipe = FluxPipeline.from_single_file(file_flux, torch_dtype=torch.bfloat16, token=huggingface_token).to(device)
 
 # Load and fuse LoRA BEFORE quantizing
 print('Loading and fusing lora, please wait...')