text-to-image-bias

Running on Zero

Avijit Ghosh commited on Jun 13, 2024

Commit

ebac435

1 Parent(s): 69de6de

Reorder models

Files changed (1) hide show

app.py CHANGED Viewed

@@ -147,11 +147,11 @@ with gr.Blocks(title="Skin Tone and Gender bias in Text to Image Models") as dem
     model_dropdown = gr.Dropdown(
         label="Choose a model",
         choices=[
             "stabilityai/sdxl-turbo",
-            "runwayml/stable-diffusion-v1-5",
             "ByteDance/SDXL-Lightning",
-            "segmind/SSD-1B",
-            "stabilityai/stable-diffusion-3-medium-diffusers"
         ],
         value=default_model
     )

     model_dropdown = gr.Dropdown(
         label="Choose a model",
         choices=[
+            "stabilityai/stable-diffusion-3-medium-diffusers",
             "stabilityai/sdxl-turbo",
             "ByteDance/SDXL-Lightning",
+            "runwayml/stable-diffusion-v1-5",
+            "segmind/SSD-1B"
         ],
         value=default_model
     )