Oct 24, 2024, 11:32 AM

app.py

@@ -26,7 +26,6 @@ API_OLD = urllib.parse.unquote(os.getenv("API_OLD"))
 # print(API_V1_MODELS)
 # print(API_V1_IMAGE)
 
-
 CACHE_DIR = os.path.join(os.path.dirname(__file__), "cache")
 IMAGE_DIR = os.path.join(CACHE_DIR, "images")
 if not os.path.exists(CACHE_DIR):
@@ -60,6 +59,7 @@ theme = gr.themes.Soft(
 
 title = "Bilder Builder"
 
+
 # MARK: GET PREV. IMAGES
 def get_gallery_images(dirpath):
     gallery_images = [dirpath + "/" + s for s in os.listdir(dirpath) if os.path.isfile(os.path.join(dirpath, s))]
@@ -150,6 +150,7 @@ def image_get_dominant_color(image_path):
 def clear_components():
     return None
 
+
 def process(Prompt, used_model, image_width, image_height, image_ratio, image_seed, randomize_seed):
     if Prompt == "":
         gr.Info("Kein Prompt angegeben, es wird ein zufälliger Prompt generiert.", duration=12)
@@ -215,7 +216,7 @@ with gr.Blocks(theme=theme, head=custom_head, css=custom_css, js=custom_js, titl
                 with gr.Row(elem_classes="image_size_selctor_wrapper"):
                     with gr.Column(scale=1):
                         with gr.Row():
-                            select_model = gr.Dropdown(choices=["FLUX", "FLUX-API-OLD", "FLUX-Realism", "FLUX-Pixel", "ANY-DARK", "Stable-Diffusion-XL-Lightning", "Stable-Diffusion-XL-Base"], value="FLUX", label="Model", elem_id="select_model", elem_classes="select-model")
+                            select_model = gr.Dropdown(choices=["FLUX-API-OLD", "FLUX", "FLUX-Realism", "FLUX-Pixel", "FLUX-Anime", "FLUX-3D", "FLUX-Disney", "ANY-DARK", "Stable-Diffusion-XL-Base", "Stable-Diffusion-XL-Lightning"], value="FLUX", label="Model", elem_id="select_model", elem_classes="select-model")
                             # with gr.Row():
                             image_width = gr.Number(label="Breite", minimum=256, maximum=MAX_IMAGE_SIZE, value=1152, step=128, elem_id="image_width_selector", elem_classes="image-width-selector", scale=1, visible=False)
                             image_height = gr.Number(label="Höhe", minimum=256, maximum=MAX_IMAGE_SIZE, value=2048, step=128, elem_id="image_height_selector", elem_classes="image-height-selector", scale=1, visible=False)