Update app.py

app.py

@@ -14,7 +14,7 @@ with open('loras.json', 'r') as f:
     loras = json.load(f)
 
 # Initialize the base model
-base_model = "black-forest-labs/FLUX.1-dev"
+base_model = "sayakpaul/FLUX.1-merged"
 pipe = DiffusionPipeline.from_pretrained(base_model, torch_dtype=torch.bfloat16)
 
 MAX_SEED = 2**32-1
@@ -110,13 +110,17 @@ css = '''
 '''
 with gr.Blocks(theme=gr.themes.Soft(), css=css) as app:
     title = gr.HTML(
-        """<h1><img src="https://huggingface.co/spaces/multimodalart/flux-lora-the-explorer/resolve/main/flux_lora.png" alt="LoRA"> FLUX LoRA the Explorer</h1>""",
+        """<h1><img src="https://huggingface.co/spaces/multimodalart/flux-lora-the-explorer/resolve/main/flux_lora.png" alt="LoRA"> SOONfactory </h1>""",
         elem_id="title",
+    )
+    	    # Info blob stating what the app is running
+    info_blob = gr.HTML(
+        """<div id="info_blob"> Activist, Futurist, and Realist LoRa-stocked Quick-Use Image Manufactory (over Flux Schnell)</div>"""
     )
     selected_index = gr.State(None)
     with gr.Row():
         with gr.Column(scale=3):
-            prompt = gr.Textbox(label="Prompt", lines=1, placeholder="Type a prompt after selecting a LoRA")
+            prompt = gr.Textbox(label="Prompt", lines=1, placeholder="Select LoRa/Style & type prompt!")
         with gr.Column(scale=1, elem_id="gen_column"):
             generate_button = gr.Button("Generate", variant="primary", elem_id="gen_btn")
     with gr.Row():
@@ -124,7 +128,7 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as app:
             selected_info = gr.Markdown("")
             gallery = gr.Gallery(
                 [(item["image"], item["title"]) for item in loras],
-                label="LoRA Gallery",
+                label="LoRA Inventory",
                 allow_preview=False,
                 columns=3,
                 elem_id="gallery"
@@ -134,15 +138,15 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as app:
             result = gr.Image(label="Generated Image")
 
     with gr.Row():
-        with gr.Accordion("Advanced Settings", open=False):
+        with gr.Accordion("Advanced Settings", open=True):
             with gr.Column():
                 with gr.Row():
-                    cfg_scale = gr.Slider(label="CFG Scale", minimum=1, maximum=20, step=0.5, value=3.5)
-                    steps = gr.Slider(label="Steps", minimum=1, maximum=50, step=1, value=28)
+                    cfg_scale = gr.Slider(label="CFG Scale", minimum=1, maximum=20, step=1, value=3)
+                    steps = gr.Slider(label="Steps", minimum=1, maximum=50, step=1, value=6)
                 
                 with gr.Row():
-                    width = gr.Slider(label="Width", minimum=256, maximum=1536, step=64, value=1024)
-                    height = gr.Slider(label="Height", minimum=256, maximum=1536, step=64, value=1024)
+                    width = gr.Slider(label="Width", minimum=256, maximum=1536, step=64, value=768)
+                    height = gr.Slider(label="Height", minimum=256, maximum=1536, step=64, value=768)
                 
                 with gr.Row():
                     randomize_seed = gr.Checkbox(True, label="Randomize seed")
@@ -162,5 +166,5 @@ with gr.Blocks(theme=gr.themes.Soft(), css=css) as app:
         outputs=[result, seed]
     )
 
-app.queue()
-app.launch()
+app.queue(default_concurrency_limit=2).launch(show_error=True)
+app.launch()