Update app.py

app.py

@@ -11,6 +11,7 @@ from peft import LoraConfig
 from PIL import Image
 from pipelines.flux_pipeline.pipeline import SynCDFluxPipeline
 from pipelines.flux_pipeline.transformer import FluxTransformer2DModelWithMasking
+from huggingface_hub import hf_hub_download
 
 HF_TOKEN = os.getenv('Syncd')
 
@@ -57,7 +58,12 @@ lora_config = LoraConfig(
     target_modules=target_modules,
 )
 pipeline.transformer.add_adapter(lora_config)
-finetuned_path = torch.load('models/pytorch_model.bin', map_location='cpu', weights_only=False)
+#finetuned_path = torch.load('models/pytorch_model.bin', map_location='cpu', weights_only=False)
+# Tự động tải file model chuẩn từ Repo gốc về
+model_path = hf_hub_download(repo_id="nupurkmr9/SynCD", filename="models/pytorch_model.bin", repo_type="space")
+
+# Load file vừa tải về
+finetuned_path = torch.load(model_path, map_location='cpu', weights_only=False)
 transformer_dict = {}
 for key,value in finetuned_path.items():
     if 'transformer.base_model.model.' in key: