Update app.py

app.py

@@ -328,12 +328,12 @@ else:
     print("Running on CPU; models loaded in float32.")
 
 # A dictionary to easily choose the model based on selection.
-DEFAULT_MODEL = "RealVisXL_V5.0_Lightning"
-MODEL_CHOICES = [DEFAULT_MODEL, "RealVisXL_V4.0", "Animagine XL 4.0"]
+DEFAULT_MODEL = "Lightning 5"
+MODEL_CHOICES = [DEFAULT_MODEL, "Lightning 4", "Animagine 4"]
 models = {
-    "RealVisXL_V5.0_Lightning": pipe,
-    "RealVisXL_V4.0": pipe2,
-    "Animagine XL 4.0": pipe3
+    "Lightning 5": pipe,
+    "Lightning 4": pipe2,
+    "Animagine 4": pipe3
 }
 
 def save_image(img, path):
@@ -465,6 +465,12 @@ with gr.Blocks(css=css, theme="bethecloud/storj_theme") as demo:
                     value="1x1",
                     label="Grid Size"
                 )
+            with gr.Row(visible=True):
+                model_selection = gr.Dropdown(
+                    choices=MODEL_CHOICES,
+                    value=DEFAULT_MODEL,
+                    label="Model Selection",
+                )
             with gr.Row(visible=True):
                 filter_selection = gr.Dropdown(
                     show_label=True,
@@ -492,12 +498,6 @@ with gr.Blocks(css=css, theme="bethecloud/storj_theme") as demo:
                     value=DEFAULT_STYLE_NAME,
                     label="Quality Style",
                 )
-            with gr.Row(visible=True):
-                model_selection = gr.Dropdown(
-                    choices=MODEL_CHOICES,
-                    value=DEFAULT_MODEL,
-                    label="Model Selection",
-                )
             with gr.Accordion("Advanced options", open=False):
                 use_negative_prompt = gr.Checkbox(label="Use negative prompt", value=True, visible=True)
                 negative_prompt = gr.Text(