Update app.py

app.py

@@ -11,8 +11,13 @@ from typing import Tuple
 dtype = torch.bfloat16
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
+# Load the base model pipeline
 pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", torch_dtype=dtype).to(device)
 
+# Load the Flux Realism LoRA model
+lora_repo = "XLabs-AI/flux-RealismLora"
+pipe.load_lora_weights(lora_repo)
+
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 2048
 
@@ -71,7 +76,7 @@ style_list = [
     },
 ]
 
-css="""
+css = """
 #col-container {
     margin: 0 auto;
     max-width: 530px;