Update app.py

app.py

@@ -7,10 +7,10 @@ import time
 import numpy as np
 import subprocess
 import torch
+import json
 from transformers import AutoProcessor, AutoModelForCausalLM
 from PIL import Image
 from deep_translator import GoogleTranslator
-import json
 from datetime import datetime
 from fastapi import FastAPI
 
@@ -124,20 +124,19 @@ examples = [
     "a beautiful woman with black hair and brown eyes",
 ]
 
-
 css = """
 #app-container {
     max-width: 930px;
     margin-left: auto;
     margin-right: auto;
 }
-".gradio-container {background: url('file=DigiP-AI/FLUX.Dev-LORA/abstract.jpg')}
+".gradio-container {background: url('file=abstract.jpg')}
    
 """
 with gr.Blocks(theme=theme, css=css, elem_id="app-container") as app:
     gr.HTML("<center><h6>🎨 FLUX.1-Dev with LoRA 🇬🇧</h6></center>")
     with gr.Tab("Text to Image"):
-        with gr.Column(elem_id=".gradio-container"):
+        with gr.Column(elem_id="app-container"):
             with gr.Row():
                 with gr.Column(elem_id="prompt-container"):
                     with gr.Row():
@@ -147,10 +146,11 @@ with gr.Blocks(theme=theme, css=css, elem_id="app-container") as app:
                         		gr.Markdown("""
                                     - **sdxl-realistic**: szn style
                                     - **stylesdxl-cyberpunk**: szn style
+                                    - **Flux-Super-Realism-LoRA**: Super Realism
                                     - **surreal-harmony**: Surreal Harmony
                                     - **extremely-detailed**: extremely detailed
                                     - **dark-fantasy**: Dark Fantasy
-                                    - **analogredmond**: AnalogRedmAF
+                                    - **Flux.1-Dev-LoRA-HDR-Realism**: HDR
                                     - **jules-bastien-lepage-style**: Jules Bastien Lepage Style
                                     - **john-singer-sargent-style**: John Singer Sargent Style
                                     - **alphonse-mucha-style**: Alphonse Mucha Style
@@ -162,16 +162,16 @@ with gr.Blocks(theme=theme, css=css, elem_id="app-container") as app:
                         		""")                       
                             
                     with gr.Row():
-                        custom_lora = gr.Dropdown([" ", "jwu114/lora-sdxl-realistic", "issaccyj/lora-sdxl-cyberpunk", "hugovntr/flux-schnell-realism", "fofr/sdxl-deep-down", "KappaNeuro/surreal-harmony", "ntc-ai/SDXL-LoRA-slider.extremely-detailed", "prithivMLmods/Canopus-LoRA-Flux-FaceRealism", "KappaNeuro/dark-fantasy", "artificialguybr/analogredmond", "KappaNeuro/jules-bastien-lepage-style", "KappaNeuro/john-singer-sargent-style", "KappaNeuro/alphonse-mucha-style", "ntc-ai/SDXL-LoRA-slider.ultra-realistic-illustration", "ntc-ai/SDXL-LoRA-slider.eye-catching", "KappaNeuro/john-constable-style", "dvyio/flux-lora-film-noir", "dvyio/flux-lora-pro-headshot"], label="Custom LoRA",)
+                        custom_lora = gr.Dropdown([" ", "jwu114/lora-sdxl-realistic", "issaccyj/lora-sdxl-cyberpunk", "strangerzonehf/Flux-Super-Realism-LoRA", "hugovntr/flux-schnell-realism", "fofr/sdxl-deep-down", "KappaNeuro/surreal-harmony", "ntc-ai/SDXL-LoRA-slider.extremely-detailed", "prithivMLmods/Canopus-LoRA-Flux-FaceRealism", "KappaNeuro/dark-fantasy", "prithivMLmods/Flux.1-Dev-LoRA-HDR-Realism", "KappaNeuro/jules-bastien-lepage-style", "KappaNeuro/john-singer-sargent-style", "KappaNeuro/alphonse-mucha-style", "ntc-ai/SDXL-LoRA-slider.ultra-realistic-illustration", "ntc-ai/SDXL-LoRA-slider.eye-catching", "KappaNeuro/john-constable-style", "dvyio/flux-lora-film-noir", "dvyio/flux-lora-pro-headshot"], label="Custom LoRA",)
                     with gr.Row():
                         with gr.Accordion("⚙️ Advanced Settings", open=False, elem_id="settings-container"):
-                            negative_prompt = gr.Textbox(label="Negative Prompt", lines=5, placeholder="What should not be in the image", value=" (visible hand:1.3), (ugly:1.3), (duplicate:1.2), (morbid:1.1), (mutilated:1.1), out of frame, bad face, extra fingers, mutated hands, (poorly drawn hands:1.1), (poorly drawn face:1.3), (mutation:1.3), (deformed:1.3), blurry, (bad anatomy:1.1), (bad proportions:1.2), (extra limbs:1.1), cloned face, (disfigured:1.2), gross proportions, malformed limbs, (missing arms:1.1), (missing legs:1.1), (extra arms:1.2), (extra legs:1.2), fused fingers, too many fingers, (long neck:1.2), sketched by bad-artist, (bad-image-v2-39000:1.3) , signature, unprompted-nsfw")
+                            negative_prompt = gr.Textbox(label="Negative Prompt", lines=5, placeholder="What should not be in the image", value=" Bad anatomy, Bad proportions, Deformed, Disconnected limbs, Disfigured, Extra arms, Extra limbs, Extra hands, Fused fingers, Gross proportions, Long neck, Malformed limbs, Mutated, Mutated hands, Mutated limbs, Missing arms, Missing fingers, Poorly drawn hands, Poorly drawn face ugly, tiling, poorly drawn hands, poorly drawn feet, poorly drawn face, out of frame, extra limbs, disfigured, deformed, body out of frame, bad anatomy, watermark, signature, cut off, low contrast, underexposed, overexposed, bad art, beginner, amateur, distorted face, blurry, draft, grainy, text, logo, watermark, banner, extra digits, signature, unprompted-nsfw")
                             with gr.Row():
                                 width = gr.Slider(label="Image Width", value=896, minimum=64, maximum=1216, step=32)
                                 height = gr.Slider(label="Image Height", value=1152, minimum=64, maximum=1216, step=32)
                             steps = gr.Slider(label="Sampling steps", value=50, minimum=1, maximum=100, step=1)
                             cfg = gr.Slider(label="CFG Scale", value=3.5, minimum=1, maximum=20, step=0.5)
-                            method = gr.Radio(label="Sampling method", value="DPM++ 2M Karras", choices=["DPM++ 2M Karras", "DPM++ SDE Karras", "DEIS", "LMS", "DPM Adaptive", "DPM++ 2M", "DPM2 Ancestral", "DPM++ S", "DPM++ SDE", "DDPM", "DPM Fast", "Euler", "Euler CFG PP", "Euler a", "Euler Ancestral", "Euler+beta", "Heun", "Heun PP2", "DDIM", "PLMS", "UniPC", "UniPC BH2"])
+                            method = gr.Radio(label="Sampling method", value="DPM++ 2M Karras", choices=["DPM++ 2M Karras", "DPM++ SDE Karras", "DEIS", "LMS", "DPM Adaptive", "DPM++ 2M", "DPM2 Ancestral", "DPM++ S", "DPM++ SDE", "DDPM", "DPM Fast", "dpmpp_2s_ancestral", "Euler", "Euler CFG PP", "Euler a", "Euler Ancestral", "Euler+beta", "Heun", "Heun PP2", "DDIM", "PLMS", "UniPC", "UniPC BH2"])
                             strength = gr.Slider(label="Prompt Strength", value=100, minimum=0, maximum=100, step=1)
                             seed = gr.Slider(label="Seed", value=-1, minimum=-1, maximum=1000000000, step=1)
                     with gr.Row():
@@ -202,16 +202,8 @@ with gr.Blocks(theme=theme, css=css, elem_id="app-container") as app:
             image_output = gr.Image(format="png")
         with gr.Row():    
             image_button = gr.Button("Run", variant='primary')
-            image_button.click(flip_image, inputs=image_input, outputs=image_output)
-
-    with gr.Tab("Information"):
-        with gr.Row():
-            gr.Markdown("""
-                                    - **Information**: 
-                                    Some of the LoRAS may not work all of the time, I don't know why, just keep trying them.
-                                    
-                        		""")
-
+            image_button.click(flip_image, inputs=image_input, outputs=image_output, concurrency_limit=None)
+            
+    app.queue()  # <-- Sets up a queue with default parameters
     if __name__ == "__main__": 
-        app.launch(show_api=False, share=False)
-        
+        app.launch(show_api=False, share=True)