Update frontend/webui/ui.py

frontend/webui/ui.py

@@ -9,11 +9,17 @@ from paths import FastStableDiffusionPaths
 from state import get_settings
 
 app_settings = get_settings()
-
-
-# Start with the models you want to load (these could come from your config)
-model_list = ["model-a", "model-b", "model-c"]
-
+pony_models = [
+    "Lykon/Pony-Diffusion",
+    "andite/anything-v4.0",
+    "stablediffusionapi/pony-lcm-turbo",
+]
+standard_models = [
+    "runwayml/stable-diffusion-v1-5",
+    "stabilityai/stable-diffusion-2-1-base",
+    "SG161222/Realistic_Vision_V5.1_noVAE",
+]
+model_array = pony_models + standard_models
 def update_model_array(text_input):
     model_list.append(text_input)
     updated = model_list