Update app.py

app.py

@@ -12,13 +12,13 @@ huggingface_hub.login(os.getenv('HF_TOKEN'))
 
 check_min_version("0.30.2")
 transformer = FluxTransformer2DModel.from_pretrained(
-        "black-forest-labs/FLUX.1-dev", subfolder='transformer', torch_dytpe=torch.bfloat16
+        "XLabs-AI/flux-RealismLora", subfolder='transformer', torch_dytpe=torch.bfloat16
     )
 
 # Build pipeline
 controlnet = FluxControlNetModel.from_pretrained("alimama-creative/FLUX.1-dev-Controlnet-Inpainting-Beta", torch_dtype=torch.bfloat16)
 pipe = FluxControlNetInpaintingPipeline.from_pretrained(
-    "black-forest-labs/FLUX.1-dev",
+    "XLabs-AI/flux-RealismLora",
     controlnet=controlnet,
     transformer=transformer,
     torch_dtype=torch.bfloat16