Upload app.py with huggingface_hub

app.py

@@ -9,12 +9,11 @@ pipe = FluxPipeline.from_pretrained(
     'black-forest-labs/FLUX.1-dev',
     torch_dtype=torch.bfloat16,
     use_safetensors=True
-)
-pipe.to('cuda')
+).to('cuda')
 
-# Load SRPO weights
-state_dict = load_file("diffusion_pytorch_model.safetensors")
-pipe.transformer.load_state_dict(state_dict, strict=False)
+# Load SRPO weights from https://huggingface.co/tencent/SRPO
+state_dict = load_file("tencent/SRPO/diffusion_pytorch_model.safetensors")
+pipe.transformer.load_state_dict(state_dict)
 
 @spaces.GPU(duration=120)
 def generate_image(