fixed

app.py

@@ -90,21 +90,20 @@ def generate(
 
     if mode == "Generate":
         file["none"] = ""
-        if model == "Ultra":
+        if model == "Stable Image Ultra (8B + workflow)":
             url = model_url["ImageUltra"]
-        elif model == "Core":
+        elif model == "Stable Image Core (2B + workflow)":
             url = model_url["ImageCore"]
             data["style_preset"] = preset
-
-        elif model == "sd3-medium":
+        elif model == "Stable Diffusion 3 Medium (2B)":
             url = model_url["StableDiffusion3"]
-            data["model"] = model
-        elif model == "sd3-large":
+            data["model"] = "sd3-medium"
+        elif model == "Stable Diffusion 3 Large (8B)":
             url = model_url["StableDiffusion3"]
-            data["model"] = model
-        elif model == "sd3-large-turbo":
+            data["model"] = "sd3-large"
+        elif model == "Stable Diffusion 3 Large Turbo (8B Turbo)":
             url = model_url["StableDiffusion3"]
-            data["model"] = model
+            data["model"] = "sd3-large-turbo"
 
         else:
             raise ValueError("Invalid model type")
@@ -189,7 +188,7 @@ css = """
 
 
 def update_style_visibility(model):
-    if model == "Core":
+    if model == "Stable Image Core (2B + workflow)":
         return gr.update(visible=True)
     else:
         return gr.update(visible=False)
@@ -285,7 +284,7 @@ with gr.Blocks(css=css, theme="NoCrypt/miku") as demo:
         with gr.Row():
             model = gr.Dropdown(
                 label="Model",
-                choices=["Ultra", "Core", "sd3-medium", "sd3-large", "sd3-large-turbo"],
+                choices=["Stable Image Ultra (8B + workflow)", "Stable Image Core (2B + workflow)", "Stable Diffusion 3 Large Turbo (8B Turbo)", "Stable Diffusion 3 Large (8B)", "Stable Diffusion 3 Medium (2B)"],
                 value="Ultra",
             )
             mode = gr.Dropdown(
@@ -444,9 +443,9 @@ with gr.Blocks(css=css, theme="NoCrypt/miku") as demo:
         [cc-by-nc-image]: https://licensebuttons.net/l/by-nc/4.0/88x31.png
         [cc-by-nc-shield]: https://img.shields.io/badge/License-CC%20BY--NC%204.0-lightgrey.svg
 
-        ## MIT Licensed Source Code
-        Portions of this work are licensed under the MIT License. For more details, please refer to the original source at  
-        [stabilityai/stable-diffusion-3-medium]: https://huggingface.co/spaces/stabilityai/stable-diffusion-3-medium
+        **MIT Licensed Source Code**
+        Portions of this work are licensed under the MIT License. For more details, please refer to the original source at:  
+        [stabilityai/stable-diffusion-3-medium](https://huggingface.co/spaces/stabilityai/stable-diffusion-3-medium) 
         """
         )
     gr.on(