Update app.py

app.py

@@ -11,12 +11,7 @@ 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-dev", torch_dtype=dtype).to(device)
-
-# Load the Flux Realism LoRA model
-lora_repo = "XLabs-AI/flux-RealismLora"
-pipe.load_lora_weights(lora_repo)
+pipe = DiffusionPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", torch_dtype=dtype).to(device)
 
 MAX_SEED = np.iinfo(np.int32).max
 MAX_IMAGE_SIZE = 2048
@@ -76,7 +71,7 @@ style_list = [
     },
 ]
 
-css = """
+css="""
 #col-container {
     margin: 0 auto;
     max-width: 530px;
@@ -218,6 +213,7 @@ with gr.Blocks(css=css, theme="bethecloud/storj_theme") as demo:
     predefined_gallery = gr.Gallery(label="## Images Sample", columns=3, show_label=False, value=load_predefined_images1())
 
     gr.Markdown("**Disclaimer/Note:**")
+    
     gr.Markdown("🍕Model used in the space <a href='https://huggingface.co/black-forest-labs/FLUX.1-schnell' target='_blank'>black-forest-labs/FLUX.1-schnell</a>. More: 12B param rectified flow transformer distilled from [FLUX.1 [pro]](https://blackforestlabs.ai/) for 4 step generation[[blog](https://blackforestlabs.ai/announcing-black-forest-labs/)] [[model](https://huggingface.co/black-forest-labs/FLUX.1-schnell)]")
     gr.Markdown("⚠️ users are accountable for the content they generate and are responsible for ensuring it meets appropriate ethical standards.")